SSD_Tensorflow识别图片目标后,类别显示为数字的问题的修改

修改了图片标记宽文字只显示数字不直观的问题:改动的地方:原始代码检测后图像分类是数字号,不能直接可读,如下

修改visualization.py
修改代码后的结果如下:

参考以下网站修改:
https://www.cnblogs.com/Osler/p/7687282.html
改后提示找不到tensorflow models的模块,然后正面的留言中有一个方法,修改后就可以了:将import tensorflow.models.SSD_Tensorflow_master.datasets.pascalvoc_2007 as pas
改为:from datasets import pascalvoc_2007 as pas
就不会出现目录问题了
但问题又来了,发现outputimage目录下生成的图片为白色,但科学显示的plt显示的图片是正常的。因为代码还不熟悉,找了半天没发现问题,最后只好从visualization.py的原代码改起,在里面添加了一个
def num2class(n):
from datasets import pascalvoc_2007 as pas
x = pas.pascalvoc_common.VOC_LABELS.items()
for name, item in x:
if n in item:
# print(name)
return name
然后再修改:class_name = str(cls_id)为:class_name = num2class(cls_id)
然后测试运行发现,生成的图片为白色的问题解决了。

猜你喜欢

转载自blog.csdn.net/SAME999ABC/article/details/82592080