目标检测之路——tensorpack+faster rcnn

一、环境问题

TensorFlow1.12和cuda版本、cudnn版本匹配问题,折腾了很久,报错信息大概是这样:

tensorflow.python.framework.errors_impl.InternalError: cuDNN launch failure : input shape([1,3,805,1093]) filter shape([7,7,3,64])
[[node conv0/Conv2D (defined at /home/**/.local/lib/python3.6/site-packages/tensorpack/models/conv2d.py:76) = Conv2D[T=DT_FLOAT, data_format="NCHW", dilations=[1, 1, 1, 1], padding="VALID", strides=[1, 1, 2, 2], use_cudnn_on_gpu=true, _device="/job:localhost/replica:0/task:0/device:GPU:0"](Pad, conv0/W/read)]]

换掉TensorFlow版本后,会出cuda相关的错,找不到cudnn.h类似的,各种切换版本,最后是TensorFlow1.14+cuda10.0+cudnn解决的。

二、代码跑通后的问题

输出的AP和Recall极不正常,几乎为零,后来发现是annotations里bbox的id有问题,coco应该是一直累加的,而我是每个图片里的各个bbox均从0开始࿰

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值