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)
然后测试运行发现,生成的图片为白色的问题解决了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值