体验新版 GitCode,发现更多精彩内容 >>
TF_INPUT_NODE=input
TF_OUTPUT_NODE=output TF_MODEL_FILE_PATH=path/to/tf_model_opt.pb MODEL_TAG=mobile_net INPUT_SHAPE=1,64,64,3 OUTPUT_SHAPE=1,64,64,2 RUNTIME=gpu TUNING_OR_NOT=1