详解Centernet训练自己的数据集出现的问题(liunx+ cuda10 + pytorch1.0.1)

ps:最近在参加华为软挑,导师还布置了一大堆任务,题也没刷几个,抽时间贴一下自己走过的错路。

训练的流程就不讲了,见下面链接:
https://blog.csdn.net/weixin_41765699/article/details/100118353

主要看里面出现错误,我们怎么解决的

1.## 错误:

cannot import name 'imread' from 'scipy.misc'
这个出现在用vocl转coco数据集的过程中,主要是因为代码太老了,scipy里移除了imread;

解决办法:from imageio import imread

2.## 错误:
第二个具体报错信息没有记录,大体是提示
Ped is not define`` 这是因为在datasets_factory文件里, 没有把你的类Ped`给导入

from .dataset.Ped import Ped

3.##错误:

KeyError: ‘Traceback (most recent call last):\n File “/home/em/anaconda3/lib/python3.5

是因为你复制了coco.py到ped.py,ped.py里少了id
详解Centernet训练自己的数据集出现的问题(liunx+ cuda10 + pytorch1.0.1)
进去看看自己少了什么。

上一篇:无人驾驶中的激光雷达(四)一种Pointpillar+CenterNet的点云目标检测网络


下一篇:3D版CenterNet: CenterPoint,小修小改也能刷爆榜单