ABlueLight
ABlueLight
我用自己的扎机器做过训练,然后对比过,map比官方的要差,我的机器比较差,batch比较小,图片尺度设置的也比较小,训练出来,map只有2.8左右。 本仓库不支持多卡训练,多尺度训练不完善,训练的话还是建议官方仓库,本仓库只是从mmedetection中抽离了一下
@yingkaining 嗯,代码应该哪里有问题。但是我的机器gpu比较差,batch size和图片尺度设置的比较小。在公司的gpu开发机上训练过,比官方仓库精度低,在达到3.4左右。
en...,训练过程中只跑训练集,暂时没有设置验证集的功能,没办法跑几轮训练再在过程中跑验证。。。
最近工作有点卷,估计没时间交流了T_T~ @jindu-uriel
看起来pytorch版本的问题
@xhhlll 我在rtx2xxx, gtx1xxx上测试过是可以的;你的显卡是什么型号,下面的连接可以参考一下:smile: [安装时要保证cuda的版本与pytorch中对应的cuda版本一致](https://blog.csdn.net/qq_30614451/article/details/111173703)
solov2没有预测bbox,是bbox-free的。但是好像也有修改solov2,让其同时预测bbox的。可以看下paddle的solov2的实现,好像是有的。
1. 对,这个并非是并行运行的,只是简单的循环。 2. 目前没发现这个问题,你测试的时候是bs增大和bs小的时候训练时间一样的吗?我没测过,感觉应该会快一些呢。我有空试试。 3. 这个不依赖mmdetection了。我目前做的这个仓库也很简陋,如果能编译mmdetection推荐使用官方的代码训练好一些。我这个算是精简版的,便于理解结构吧QwQ ... ...
好像是的,感觉这里我的代码有问题,后面尝试改一下。
没有~,目前工作比较忙没有时间维护~