arg_parser.py

#
#  Original code (C) Copyright EdgeCortix, Inc. 2022
#  Modified Portion (C) Copyright Renesas Electronics Corporation 2022
#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.

from optparse import OptionParser
import os

def get_args():
    """
    Check argments and return option dictionaly data & input model file name
      Input : None (arguments)
      Output : option dictionary , AI model file path
    """
    usage = "usage: %prog [options] model_file"
    parser = OptionParser(usage)
    parser.set_defaults(drp_compiler_dir='./drp-ai_translator_release', drp_compiler_version='100', toolchain_dir='/opt/poky/3.1.31', disable_concat = False, quantization_tool='./drpai_quant', qat=False, qat_type="pytorch", num_frame=1, cpu_data_type="float16")
    parser.add_option("-r", "--record_dir", dest="record_dir", help="Calibration data record directory", metavar="DIR")
    parser.add_option("-d", "--drp_compiler_dir", dest="drp_compiler_dir", help="DRP-AI Translator root directory", metavar="DIR")
    parser.add_option("-t", "--toolchain_dir", dest="toolchain_dir", help="Cross-compilation toolchain root directory", metavar="DIR")
    parser.add_option("-o", "--output_dir", dest="output_dir", help="Output directory", metavar="DIR") # Pattern 2
    parser.add_option("-s", "--input_shape", dest="input_shape", help="User specified input shape")
    parser.add_option("-i", "--input_name", dest="input_name", help="Input name")
    parser.add_option("--level",dest="level", default=1, help="Optimization level[0 or 1]")
    parser.add_option("-q", "--fp16", action="store_true", dest="fp16", help="Convert to FP16")
    parser.add_option("-n", "--num_frame", dest="num_frame", default=1, help="Number input frames")
    parser.add_option("-c", "--quantization_tool", dest="quantization_tool", help="Quantization tool directory", metavar="DIR")
    parser.add_option("-f", "--cpu_data_type", dest="cpu_data_type", default="float16", help="Specify cpu data type (float16/float32)", metavar="float16 or float32")
    parser.add_option("--images", dest="image_dir", help="Specifies the directory where calibration images are contained.", metavar="DIR")
    parser.add_option("-v", "--drp_compiler_version", dest="drp_compiler_version", help="DRP-AI Translator version (091 or 100)", metavar="VERSION ID")
    parser.add_option("-p", "--quantization_option", dest="quantization_option", default="", help="drpai quantization option, example -p \"-az\"", metavar="OPTION")
    parser.add_option("-a", "--qat", action="store_true", dest="qat", default=False, help="Option to select compile from exported QAT model", metavar="OPTION")
    parser.add_option("-m", "--qat_type", dest="qat_type", help="QAT type", metavar="QAT TYPE")


    (options, args) = parser.parse_args()
    opts = vars(options)
    if len(args) != 1:
        parser.error("Expected a model file")

    print("[Check arguments]")
    model_file = str(args[0])
    print("  Input AI model         : ", model_file)
    
    # Set Cross-compiler path setting
    try: # From linux environment variable
        cc_path = os.environ["SDK"]
        opts["toolchain_dir"] = cc_path
        print("  SDK path               : ", opts["toolchain_dir"])
    except:
        print("  SDK path               : ", opts["toolchain_dir"], " (*default)")
    
    # Set DRP-AI Transaltor path setting
    try: # From linux environment variable
        drp_path = os.environ["TRANSLATOR"]
        opts["drp_compiler_dir"] = drp_path
        print("  DRP-AI Translator path : ", opts["drp_compiler_dir"])
    except:
        print("  DRP-AI Translator path : ", opts["drp_compiler_dir"], " (*default)")
    print("  Output dir             : ", opts["output_dir"])


    input_shape = opts["input_shape"]
    if input_shape is not None:
        input_shape = tuple([int(dim) for dim in input_shape.split(',')])
        print("  Input shape            : ", input_shape)
    else:
        input_shape = [1, 3, 224, 224]
        print("  Default input shape    : ", input_shape)
    opts["input_shape"] = input_shape

    # Set optimize level
    opts["disable_concat"] = False
    if(opts["level"]=="0" or opts["level"]==0):
        print("  Optimization level : LOW")
        opts["disable_concat"] = True
    
    return opts, model_file

if __name__ == "__main__":
    opts, model_file = args_set()