pytorch-yolov4-deepsort
pytorch-yolov4-deepsort copied to clipboard
training yolo-v4 detector
Thanks for sharing this useful repository! I'm trying to retrain the detector with my own data, in which format should I restructure my data into?
Particularly, how to construct the train.txt file to be compatible with your code?
Thank you,
- Use the VOC format for training.
- Before training, put the label file in the Annotation under the VOC2007 folder under the VOCdevkit folder.
- Before training, put the image files in JPEGImages under the VOC2007 folder under the VOCdevkit folder.
- Use the voc2yolo4.py file to generate the corresponding txt before training.
- Run voc_annotation.py in the root directory again. Before running, you need to change the classes to your own classes. Be careful not to use Chinese labels and no spaces in the folder!
- At this time, the corresponding 2007_train.txt will be generated, and each line corresponds to its picture position and its real frame position.