运行Faster-RCNN-TensorFlow-Python3.5的train.py文件出现 assert (boxes[:, 2] >= boxes[:, 0]).all()错误

参考这篇文章:https://blog.csdn.net/10km/article/details/64641322
我觉得我的xml文件不需要将矩形框都减1,所以在-1前面都加一个井号,将-1都注释掉。
错误:File "/home/drl/new/Faster-RCNN-TensorFlow-Python3.5/lib/datasets/imdb.py", line 119, in append_flipped_images assert (boxes[:, 2] >= boxes[:, 0]).all()
在这里插入图片描述解决办法是:将/home/drl/new/Faster-RCNN-TensorFlow-Python3.5/lib/datasets/pascal_voc.py打开,将162-165行代码后面的-1都注释掉!!!
在这里插入图片描述在重新运行train.py文件时,首先
1、同时将/home/drl/new/Faster-RCNN-TensorFlow-Python3.5/default/voc_2007_trainval/文件夹下的所有模型文件备份,然后删掉
2、还要将/home/drl/new/Faster-RCNN-TensorFlow-Python3.5/data/cache/文件夹下的缓存文件备份,然后删掉

再运行python train.py

在这里插入图片描述在这里插入图片描述
成功!

训练完,现在运行demo.py文件,在命令行输入代码 python demo.py
出现错误:

  File "/home/drl/anaconda3/envs/tensorflow-gpu/lib/python3.5/site-packages/tensorflow/python/framework/ops.py", line 1204, in __init__
    self._traceback = self._graph._extract_stack()  # pylint: disable=protected-access

InvalidArgumentError (see above for traceback): Assign requires shapes of both tensors to match. lhs shape= [21] rhs shape= [2]
	 [[Node: save/Assign_2 = Assign[T=DT_FLOAT, _class=["loc:@vgg_16/cls_score/biases"], use_locking=true, validate_shape=true, _device="/job:localhost/replica:0/task:0/gpu:0"](vgg_16/cls_score/biases, save/RestoreV2_2/_5)]]

在这里插入图片描述在这里插入图片描述错误原因是:
1.没有将demo.py, line 33行的代码改成自己的类别:
在这里插入图片描述
2. 没有将demo.py,line 33行的代码需要测试的图片名称改为自己需要测试的图片名称:
在这里插入图片描述
改为:

在这里插入图片描述
3. 没有在/home/drl/new/Faster-RCNN-TensorFlow-Python3.5/data/demo/文件夹下放置自己需要测试的图片

在这里插入图片描述
改为:

在这里插入图片描述然后重新运行demo.py文件:

在这里插入图片描述

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值