PyTorch_YOLOv4 icon indicating copy to clipboard operation
PyTorch_YOLOv4 copied to clipboard

Published 20 hours ago • verified publisher icon WongKinYiu

Training on Custom Dataset

Open NazmulTakbir opened this issue 4 years ago • 1 comments

Dear Wong,

I wanted to train YOLOv4pacsp-x-mish for my custom dataset. Can you please guide me in this aspect.

  1. Which pretrained weights file should I use for transfer learning? I need the yolov4-pacsp-x-mish.pt file right? But I can't seem to find it. I can only find the yolov4-pacsp-x-mish.weights file
  2. I know I have to provide my custom .yaml file. Is there anything else that I also need to do?

NazmulTakbir avatar Nov 28 '20 04:11 NazmulTakbir

Hey @NazmulTakbir

Did you try loading darknet .weights before training the model the model? Did it work?

anantgupta129 avatar Dec 23 '21 03:12 anantgupta129