search

yatengLG / Retinanet-Pytorch

Retinanet目标检测算法(简单,明了,易用,全中文注释,单机多卡训练,视频检测)(based on pytorch,Simple, Clear, Mutil GPU)
267 stars 96 forks source link

可不可以放几张效果图呢？或者在coco数据集上的map #4

Open yangbisheng2009 opened 4 years ago

yangbisheng2009 commented 4 years ago

根据我的实验结果，最终效果不太理想呢，是不是有些细节需要调优

yatengLG commented 4 years ago

我这个项目是直接改过来的，没有进行完整的训练。所以在训练参数方面可能有些出入。

你是在coco上训练的哈？

我今天抽空上传个在coco上已经训练好的模型吧。到时候你再试一试。

yangbisheng2009 commented 4 years ago

我在voc2007训练的，效果不甚理想

yatengLG commented 4 years ago

voc的没跑完，最终结果说实在话，我也不清楚了。在训练时，你可以改下参数什么的。这个参数是直接训练VGG_ssd 时候的训练参数。我只是跑了下验证了下代码是否有效，并没有进行完整训练。

在训练基于resnet50的retinanet的时候，你可能需要调整训练参数。