Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

【飞桨黑客松 Hackathon + No.153】Yolov6 v3.0 论文复现 #114

Closed

Conversation

Yang-Changhui
Copy link

@Yang-Changhui Yang-Changhui commented Mar 29, 2023

目前修复了一些训练时的函数的错误调用,已经可以进行训练,
通过比对训练日志,发现loss有些不同,目前还在寻找原因
配置文件运行:/configs/yolov6/yolov6_s_300e_coco.yml
训练3epoch的结果:
3-epoch

@nemonameless
Copy link
Collaborator

#104

@Yang-Changhui
Copy link
Author

修改了task_aligned_assigner,优化器,权重初始化部分,目前训练20epoch结果:
0cb59e5bc1958c24353806f447655d3

@nemonameless nemonameless changed the title Yolov6 v3.0 论文复现 【飞桨黑客松 Hackathon + No.153】Yolov6 v3.0 论文复现 Apr 14, 2023
@Yang-Changhui
Copy link
Author

更新了EffiDeHead_fuseab代码

@Yang-Changhui
Copy link
Author

修改后训练20epoch的结果,map一段时间下降还没解决:
image

@Yang-Changhui
Copy link
Author

目前训练23个epoch的结果,需要全量训练:
image

@Yang-Changhui
Copy link
Author

训练300epoch的结果为map 36.2,torch版本yolov6版本精度为map36.9,相差0.7;当使用预热损失的时候,精度会有提高
无标题

@Yang-Changhui
Copy link
Author

使用PaddlePaddle2.4复现了Yolov6n模型的完整训练精度,coco数据集36.2和原作者相差0.7
链接:https://pan.baidu.com/s/1j8dGh91uQuuRmkVgnVLkIg?pwd=2p5j
提取码:2p5j

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants