YOLOV6 HEAD tal

[lihualilee]

Before Asking

• [X] I have read the README carefully. 我已经仔细阅读了README上的操作指引。

• [X] I want to train my custom dataset, and I have read the tutorials for training your custom data carefully and organize my dataset correctly; (FYI: We recommand you to apply the config files of xx_finetune.py.) 我想训练自定义数据集，我已经仔细阅读了训练自定义数据的教程，以及按照正确的目录结构存放数据集。（FYI: 我们推荐使用xx_finetune.py等配置文件训练自定义数据集。）

• [X] I have pulled the latest code of main branch to run again and the problem still existed. 我已经拉取了主分支上最新的代码，重新运行之后，问题仍不能解决。

Search before asking

• [X] I have searched the YOLOv6 issues and found no similar questions.

Question

有个问题想请教一下，在以前的yolo算法中基本都是三分支的结构，分类分类用于评估分类的好坏，置信度分支用于评估框的好坏，并且在NMS环节会耦合分类的结果和置信度的结果再做NMS。在yolov6的head结构中，tal中隐式的对齐了分类分支和置信度分支，因此可以将两个分支可以合二为一，相当于隐式的做了一个对齐，置信度分支的结果其实并没有显示的表达出来，实际测试一些数据时，相较于yolox而言，yolov6的误检会多一点

Additional

No response

[LeeWise9]

same problem!

[mtjhl]

可以通过提高置信度阈值来降低 FP，也可以在训练时候，在交叉熵损失函数内，增大负样本的权重。

[joyyang1215]

我也有這個疑問, 為什麼YOLOv6把cls head, reg head和obj head合併為 cls head和reg head, 刪除了obj head? 這樣做的目的與好處是什麼?