david8862 / keras-YOLOv3-model-set

end-to-end YOLOv4/v3/v2 object detection pipeline, implemented on tf.keras with different technologies
MIT License
639 stars 221 forks source link

yolov4 loss #202

Open guishilike opened 3 years ago

guishilike commented 3 years ago

yolov4的loss损失设计和原先v3还是略有不同的,像obj梯度,iou_thresh机制等 image image

david8862 commented 3 years ago

@guishilike sorry for response delay. 本项目实现的yolov4 loss基本还是以qqwweee版本yolov3 loss为基础进行的修改,所以和原版是会有一些差别的

guishilike commented 3 years ago

实际上,在You Only Look Once中是这样说的: image yolov5中似乎也是这样实现的: image 原版darknet我看了好像确实不是。

david8862 commented 3 years ago

@guishilike 使用iou作为confidence score在YOLOv1和v2中曾经使用过,本项目的yolo2部分也有所涉及,但在YOLOv3中没有继续使用。YOLOv5的不同子版本中似乎也有过不同的实现方式