class OnnxChainer::CLI

Public Class Methods

new(argv) click to toggle source
# File lib/onnx-chainer/cli.rb, line 14
def initialize(argv)
  @argv = argv.dup
  @parser = OptionParser.new do |opts|
    opts.banner = 'onnx-red-chainer [OPTIONS] FILE'
    opts.version = VERSION
    opts.on('-o', '--output_dir=OUTPUT_DIR', 'output path') { |v| @output = v }
    opts.on('-m', '--model_name=MODEL_NAME', 'Model name') { |v| @model_name = v }
    opts.on('-h', '--help', 'show help') { @help = true }
  end

  @parser.parse!(argv)
  @onnx_path = argv.pop
end
start(argv) click to toggle source
# File lib/onnx-chainer/cli.rb, line 10
def self.start(argv)
  new(argv).run
end

Public Instance Methods

run() click to toggle source
# File lib/onnx-chainer/cli.rb, line 28
def run
  if @help || @argv.empty?
    puts @parser.help
  else
    graph = OnnxChainer.parse_file(@onnx_path)
    graph.export(output_dir: @output, model_name: @model_name)
  end
end