TFODAPI -模型微调不起作用

Question

使用Tensorflow对象检测API，我发现每个检测到的对象报告的置信度都超过50%。我希望看到较低分值的检测结果，因为在我正在处理的一些图像中，许多相关目标(人和/或汽车)被遗漏了。我也想改变score_threshold的值来提高网管速度(根据https://becominghuman.ai/tensorflow-object-detection-api-basics-of-detection-2-2-28b348495eec )。

我使用的是"ssd_mobilenet_v1_coco_2017_11_17“模型，所以我在object_detection\samples\configs中更改了"ssd_mobilenet_v1_coco.config”：

"matched_threshold“和"unmatched_threshold”从0.5到0.3，

"score_threshold“，从1e-8到0.2

而且还

"iou_threshold“，从0.6到0.8

然后，从"object_detection“文件夹中运行：

“python export_inference_graph.py --输入类型目录--管道配置路径.\samples\configs\ssd_mobilenet_v1_coco.config --训练式检查点前缀.\fine_tuned_model --输出目录.\fine_tuned_model”

然后我就得到了"frozen_inference_graph.pb“我还可以准备相关的"frozen_inference_graph.pbtxt“，在那里我看到iou_threshold实际上从0.6更改为0.8，所以我想其他参数也正确地更改了。

然后，为了使用新模型，我更改了object_detection_tutorial.ipynb :我将"MODEL_NAME = 'ssd_mobilenet_v1_coco_2017_11_17'“更改为"MODEL_NAME = 'fine_tuned_model'”，然后一步一步地执行它，跳过“下载模型”部分。我确实看到了'ssd_mobilenet_v1_coco_2017_11_17‘的不同之处，因为在狗的"image1.jpg“中，我看到右边的狗周围有两个边界框，我想是因为iou_threshold的值更高。我不能检测到置信度分数低于50%的物体。它与测试图像不是很相关，但更改图像时，我看到了许多缺失的对象。我想我错过了什么，但我不能弄清楚是什么。

谢谢!

Answer 1

我终于找到了我疑惑的答案。

很简单，只需将min_score_thresh值指定为"visualize_boxes_and_labels_on_image_array“函数的参数！:O:O:O