测试自己训练的模型及修改代码(tensorflow modles zoo)

打开anaconda后打开jupyter加载 object_detection_tutorial.ipynb

 修改1：换模型路径以及分类模型标签

图中红色实线1表示自己训练模型的路径

图中红色实线2表示自己训练生成的模型

图中红色实线3表示检测目标的标签

修改2：注释这部分代码

# download model
opener = urllib.request.URLopener()
# 下载模型，如果已经下载好了下面这句代码可以注释掉
opener.retrieve(DOWNLOAD_BASE + MODEL_FILE, MODEL_FILE)
tar_file = tarfile.open(MODEL_FILE)
for file in tar_file.getmembers():
    file_name = os.path.basename(file.name)
    if 'frozen_inference_graph.pb' in file_name:
        tar_file.extract(file, os.getcwd())

 修改3：修改红色下划线代码，检测图像的路径

 修改4：注释红色圈中代码（pycharm没有见到显示图片），增加红色下划线的代码。