YunYang1994 / tensorflow-yolov3

🔥 TensorFlow Code for technical report: "YOLOv3: An Incremental Improvement"
https://yunyang1994.gitee.io/2018/12/28/YOLOv3-算法的一点理解/
MIT License
3.63k stars 1.36k forks source link

求救!大家有基于作者的代码实现完整的yolov3-tiny吗? #353

Open ootzfoo opened 5 years ago

ootzfoo commented 5 years ago

我基于作者的代码实现了yolov3-tiny,基本上就是把pred_sbbox去掉,然后训练voc数据集。 1、把107-109移到79行,得到模型是34.9M,然后evaluate时报错,Keyconv0/batch_normalization/beta/ExponentialMovingAverage not found in checkpoint...然后把self.saver = tf.train.Saver(ema_obj.variables_to_restore())改为self.saver = tf.train.Saver(),最终map34.37% 2、用以上模型作为预训练模型,把那几行移回原本的位置,得到的模型是139.9M,然后evaluate里改回原来的,最终map39.48% 3、用AB版yolov3-tiny训练voc,最终map56.66% 4、darknet作者给出的数据是58.4% 5、yolo nano给出的数据是模型4Mmap69.1% 所以,伸手党来问一下 1)基于作者代码实现的yolov3-tiny(感觉自己的有点问题) 2)有没有yolo nano相关的资料(那篇论文除外) 谢谢大家。

KoapT commented 5 years ago

代码拿来看看噻

ootzfoo commented 5 years ago

@KoapT https://github.com/ootzfoo/TensorFlow-yolov3-tiny

chengyequn commented 5 years ago

你好,我也遇到你同样的问题,请问一下你解决了吗

LeRoii commented 4 years ago

我基于作者的代码实现了yolov3-tiny,基本上就是把pred_sbbox去掉,然后训练voc数据集。 1、把107-109移到79行,得到模型是34.9M,然后evaluate时报错,Keyconv0/batch_normalization/beta/ExponentialMovingAverage not found in checkpoint...然后把self.saver = tf.train.Saver(ema_obj.variables_to_restore())改为self.saver = tf.train.Saver(),最终map34.37% 2、用以上模型作为预训练模型,把那几行移回原本的位置,得到的模型是139.9M,然后evaluate里改回原来的,最终map39.48% 3、用AB版yolov3-tiny训练voc,最终map56.66% 4、darknet作者给出的数据是58.4% 5、yolo nano给出的数据是模型4Mmap69.1% 所以,伸手党来问一下 1)基于作者代码实现的yolov3-tiny(感觉自己的有点问题) 2)有没有yolo nano相关的资料(那篇论文除外) 谢谢大家。

你好, 我试了你的代码,voc测试集上mAP 50.4%,epochs设了40+80