WongKinYiu / ScaledYOLOv4

Scaled-YOLOv4: Scaling Cross Stage Partial Network
GNU General Public License v3.0
2.02k stars 573 forks source link

Table 13: YOLOv4-tiny (3l) model #25

Open JundongYF opened 3 years ago

JundongYF commented 3 years ago

Table 13: YOLOv4-tiny 和YOLOv4-tiny(31)的区别是什么?

WongKinYiu commented 3 years ago

it is 3l, not 31. there are 2 yolo layers in yolov4-tiny, while 3l has 3.

winterxx commented 3 years ago

where can i find YOLOv4-tiny(3l) weights?

casper-hansen commented 3 years ago

@WongKinYiu I have 2 questions since I am currently optimizing speed for a custom dataset.

  1. Can you provide cfg for yolov4-tiny with 3 yolo layers?
  2. Are the weights available for the 3-layer version? If not, should I just train it on COCO for 600 epochs as stated in the paper?

Thank you in advance!

casper-hansen commented 3 years ago

@WongKinYiu I have 2 questions since I am currently optimizing speed for a custom dataset.

  1. Can you provide cfg for yolov4-tiny with 3 yolo layers?
  2. Are the weights available for the 3-layer version? If not, should I just train it on COCO for 600 epochs as stated in the paper?

Thank you in advance!

This repo does not have any of these available, so the results cannot be reproduced solely based on this repository. However, I found the answer to both questions in the Darknet repo. I haven't tested them out, but here they supposedly are:

  1. The 3l config: https://github.com/AlexeyAB/darknet/blob/master/cfg/yolov4-tiny-3l.cfg
  2. The 3l weights: https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v4_pre/yolov4-tiny.conv.29

I found the weights from AlexeyAB in this thread

https://github.com/AlexeyAB/darknet/issues/7087#issuecomment-745549924