TensorFlow下DOTA预训练SSD目标检测模型结果可视化无输出问题

问题可能原因 #

• 缺少matplotlib展示触发语句：你提供的代码中仅调用了plt.imshow(image_np)，没有追加plt.show()语句，matplotlib在非交互运行模式下不会主动渲染显示图片，添加该行即可触发弹窗展示可视化结果。
• Jupyter环境缺少魔法声明：如果在Jupyter Notebook/Lab中运行代码，需要在代码块开头添加%matplotlib inline声明，才能让可视化结果直接输出在单元格中。
• 检测置信度阈值过高：vis_util.visualize_boxes_and_labels_on_image_array默认仅绘制置信度≥0.5的检测框，若模型本次输出的所有检测结果得分都低于该阈值，就不会有任何标注绘制在图上。可以传入参数min_score_thresh=0.2降低阈值，验证是否有低置信度检测结果。
• 无图形界面环境后端不兼容：如果在无图形界面的远程服务器上运行代码，matplotlib默认使用非交互式的Agg后端，无法生成GUI展示窗口。可以添加plt.savefig(f"detect_result_{image_path.split('/')[-1]}")语句将结果图直接保存到本地磁盘，再下载查看。
• 类别映射配置错误：检查传入的category_index是否和模型输出的检测类别编号匹配，若映射关系为空或不匹配，也会导致标注绘制失败。

内容的提问来源于stack exchange，提问作者porschewitz