Tianxiaomo / pytorch-YOLOv4

PyTorch ,ONNX and TensorRT implementation of YOLOv4
Apache License 2.0
4.48k stars 1.49k forks source link

generate new anchors for new dataset #300

Open xmba15 opened 3 years ago

xmba15 commented 3 years ago

Hi @Tianxiaomo Thank you for the code. I would like to consult a question. As far as I know, yolov4 still uses the anchor-based object-detection approach like yolov3. I have looked through the train.py script in this repository and it seems to me that you use fixed anchors in Yolo_loss function? Is yolov4 neglecting the need of generating dataset-specific anchors? Or else, do you intend to add k-means-based anchor generation function?

TA-Robot commented 3 years ago

Is there any answer?