thundernet

因为faster_rcnn太慢了。转头去实验thundernet。

网上有大神给了一个非官方版本。https://github.com/ouyanghuiyu/Thundernet_Pytorch

代码结构上,和最近看到的faster_rcnn的结构非常像,很多地方代码也复用了,看来这是两阶段物体检测里比较被认可的代码结构。

问题嘛,非常的多,非官方版本,很多地方我都搞不懂在干什么。

demo.py运行,scipy.misc.imread注释掉。

底下的imread,都用cv2.imread代替,这边代码要修改下,然后把通道转换的代码,,我还没有确认下,,

然后找不到nms,需要参考https://github.com/ouyanghuiyu/Thundernet_Pytorch/issues/27。

然后我个人还遇到一个问题,import的时候,sys.path里面有上一个rcnn的lib目录,导致import都错了,找了一下没有找到很好的全局修改sys.path的方案,就每次在代码的最前面对sys.path进行修改,把thundernet的目录加进来。

不行,这个项目只支持gpu的,cpu only没办法work。大家有兴趣的自己去调试吧,作者还有个类似的在mmdetecction框架下实现的也可以尝试下。

原文地址:https://www.cnblogs.com/zherlock/p/13157311.html