Sha256: ef5779db38b4cbc6252aefc6525978f0cac7226d0728a50ba41aa9838536c5aa
Contents?: true
Size: 905 Bytes
Versions: 2
Compression:
Stored size: 905 Bytes
Contents
# frozen_string_literal: true require 'onnx-chainer' require 'json' require 'optparse' require 'pathname' module OnnxChainer class CLI def self.start(argv) new(argv).run end 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 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 end end
Version data entries
2 entries across 2 versions & 1 rubygems
Version | Path |
---|---|
onnx-red-chainer-0.1.2 | lib/onnx-chainer/cli.rb |
onnx-red-chainer-0.1.1 | lib/onnx-chainer/cli.rb |