DataXujing/YOLOv9

train.py 和 train_dual.py 的区别是什么?

Closed this issue · 2 comments

train.py 和 train_dual.py 的区别是什么?

可以参考这个解释:WongKinYiu/yolov9#1
train_dual.py是有一个辅助分支和一个主分支的训练代码。

train.py: no aux branch + one main branch
train_dual: one aux branch + one main branch.
triple_branch: two aux branches + one main branch

We provide these codes to show how to extend aux branches for training. Everyone could follow the rule to make related researches. For example, we also extend train_multiple to implement training strategy which is similar to [Co-DETR](https://openaccess.thecvf.com/content/ICCV2023/html/Zong_DETRs_with_Collaborative_Hybrid_Assignments_Training_ICCV_2023_paper.html).

谢谢博主