Open saikrishnadas opened 3 years ago
The main advantage of Scaled-Yolo was its speed. I tried training a p7 model with 2 classes with 500 images each (total 1000 images) , 64 batch-size and 1000 epochs, but i took me around 20hrs to complete the training process which is slower than darknet yolov4 model. Why is it so ? I use single Nvidia tesla GPU.
it because p7 model is 5 times larger than yolov4 model. for similar model scale, please use yolov4-csp.
Okay, And why do i get multiple predictions on a single object? I trained plum and apple as a model and while inference I get multiple predictions on a single object inside the image.
Attaching the images,
the one in the top gives 2 predictions
So, you recommend me to use the best and not last? But sometimes best comes with at 500 epochs from 2000 epochs. !! And how do I calculate MAP for each weight after the training is completed?
Which one is suitable for final production? And what's the difference between a normal and strip weight?
Where do I change the loss function? I wanted to use Cross entropy instead of binary CE , where do i change it ?
https://github.com/WongKinYiu/ScaledYOLOv4/blob/yolov4-csp/data/hyp.scratch.yaml#L17-L19