tensorflow-yolov3 icon indicating copy to clipboard operation
tensorflow-yolov3 copied to clipboard

Published 20 hours ago • verified publisher icon YunYang1994

请问如何使用自己训练的权重进行预测

Open ganyibo opened this issue 5 years ago • 1 comments

当我把video_demo.py中权重加载路径改成自己训练的权重文件之后进行预测的时总是会报错如下: tensorflow.python.framework.errors_impl.NotFoundError: /home/GanYB/PycharmProjects/object_detection/tensorflow-yolov3/checkpoint/yolov3_train_loss=6.1510.ckpt-50; No such file or directory 请问这个问题怎么解决呢

ganyibo avatar Aug 01 '20 13:08 ganyibo

__C.TEST.WEIGHT_FILE = "./checkpoint/yolov3_test_loss=20.6818.ckpt-26" 去config.py修改这行参数，选择最小loss的就行

Atange1998 avatar Jan 30 '21 06:01 Atange1998