search

hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite
https://github.com/hunglc007/tensorflow-yolov4-tflite
MIT License
2.24k stars 1.24k forks source link

[Feature Request] Enable the possibility of custom anchors #320

Open aparico opened 3 years ago

aparico commented 3 years ago

Why? A lot of people are having issues on this repo because they trained their models with custom anchors. It becomes complicated for people to change the source code to enable custom anchors.

What to consider?

  1. Avoid anchor_per_scale --> Sometimes, it's not the 3,3,3. For example, it can be 4,3,2.
  2. Enable inputting custom anchors