YOLOv5 ver 6 model included BiFPN? #5328
-
|
I notice there is in the models/hub yolov5-bifpn.yaml. is the latest yolov5 model using BiFPN then PANet? |
Beta Was this translation helpful? Give feedback.
Replies: 2 comments
-
|
@ys31jp yes, there is a YOLOv5-BiFPN model in hub. This is a v5.0 model. We trained this model to match the BiFPN head of EfficientDet, but did not see substantial mAP gains with this change, and we learned that the current YOLOv5 PANet head is actually very similar to what the EffiicentDet authors call a BiFPN head.
|
Beta Was this translation helpful? Give feedback.
-
|
Thanks for replied. Great to know this. Will check on the similarity for the BiFPN on the model. |
Beta Was this translation helpful? Give feedback.
@ys31jp yes, there is a YOLOv5-BiFPN model in hub. This is a v5.0 model. We trained this model to match the BiFPN head of EfficientDet, but did not see substantial mAP gains with this change, and we learned that the current YOLOv5 PANet head is actually very similar to what the EffiicentDet authors call a BiFPN head.
https://github.com/ultralytics/yolov5/blob/master/models/hub/yolov5-bifpn.yaml