PaulChongPeng / YOLO2TensorFlow

YOLOv2 implemented by TensorFlow
Apache License 2.0
38 stars 18 forks source link

我看这个网络直接输出分类、框坐标、置信度,并没有混合分类和检测数据集,对分类和检测任务进行联合训练 #11

Open jiashihuigithub opened 5 years ago

jiashihuigithub commented 5 years ago

YOLO2提出一种联合训练机制,混合来自检测和分类数据集的图像进行训练。当网络看到标记为检测的图像时,基于完整的yolov2损失函数进行反向传播。当它看到一个分类图像时,只从特定于分类的部分反向传播损失。

jiashihuigithub commented 5 years ago

论文中的darknet19图只是用来做分类的网络结构,然后添加了一些层做目标检测网络,我可以这样理解么?我看代码中目标检测网络并没有像论文中说的那样,添加3个11和33,是我理解有问题么?