我尝试使用ssd_mobilenet_v1_coco_11_06_2017
,ssd_mobilenet_v1_coco_2018_01_28
和faster_rcnn_inception_v2_coco_2018_01_28
训练模型,但是当我尝试运行object_detection_tutorial.ipynb
并测试{{1} }我得到的只是没有边界框的图像,我使用test_images
训练了模型,还尝试了model_main.py
,但两者均损失了<1。我正在使用train.py
,并在tensorflow = 1.14
上尝试过。我陷入了最后一步。我很肯定我正确创建了tensorflow = 2.0
。并且当我运行模型(tfrecords
,ssd_mobilenet_v1_coco_11_06_2017
和ssd_mobilenet_v1_coco_2018_01_28
)时,我对它们进行了训练,它们可以正常工作,因此我怀疑我的模型出了问题
答案 0 :(得分:0)
您是否在object_detection_tutorial.ipynb文件中更新了模型的路径,“ tf.saved_model.load”是您必须提供经过训练的模型的路径的API。