nolog_caffe_predict_top2_scroces_bot_gpu.py 文件源码

def main(argv):
    #input e.g : python caffe_main_test_top2_scores.py ./test_dir/ ./0908.log
    #get argvs
    caffe_root = '/opt/caffe/'
    model_path = '/opt/model_0908_12c/'
    parser = argparse.ArgumentParser()
        # Required arguments: input and output files.
        parser.add_argument(
        "image_path",
        help="Input image, directory, or npy."
        )
        parser.add_argument(
        "log_path",
        help="Output npy filename."
        )
        args = parser.parse_args()
        #set vars from argvs
        image_path = args.image_path
        log_path = args.log_path
    #program begin...
    #caffe.set_mode_cpu()