使用Tensorflow对象检测API在运行eval.py脚本时如何更改IOU

时间:2019-05-10 13:53:09

标签: python tensorflow deep-learning computer-vision object-detection-api

我已经在自己的自定义数据集中使用TensorFlow对象检测API训练了FasterRCNN_Inception_Resnet_v2模型。我能够成功评估性能,并获得将IOU设置为0.5的每个类的AP。但是,我无法将IOU更改为其他值(例如0.75)。

我尝试将文件object_ detection/utils/object_detection_evaluation.py中出现的所有出现的0.5更改为0.75。但是,即使我运行python eval.py --logtostderr --pipeline_config_path=training/faster_rcnn_inception_resnet_v2_atrous_pets.conf --checkpoint_dir=training --eval_dir=eval,mAP结果仍然显示为0.5 IOU。

0 个答案:

没有答案