search

tjuskyzhang / Scaled-YOLOv4-TensorRT

Got 100fps on TX2. Got 500fps on GeForce GTX 1660 Ti. If the project is useful to you, please Star it.
178 stars 41 forks source link

Yolov4 Tiny Anchors wrong? #19

Closed philipp-schmidt closed 3 years ago

philipp-schmidt commented 3 years ago

Hi, in cfg the anchors for tiny yolo are:

anchors = 10,14, 23,27, 37,58, 81,82, 135,169, 344,319

In code they are:

{81,82, 135,169, 344,319}
{23,27, 37,58, 81,82}

So this is probably wrong, the anchor 81,82 is used twice and 10,14 is never used:

https://github.com/tjuskyzhang/Scaled-YOLOv4-TensorRT/blob/7aa4e2bad63caf79beec4c60246e77476e018df7/yolov4-tiny-tensorrt/yololayer.h#L36

philipp-schmidt commented 3 years ago

Nevermind I was proven wrong.

https://github.com/hunglc007/tensorflow-yolov4-tflite/issues/111#issuecomment-650205659