search

WongKinYiu / yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)
GNU General Public License v3.0
1.98k stars 524 forks source link

How do we train yolor with custom dataset? #8

Open thanhnguyentrung opened 3 years ago

thanhnguyentrung commented 3 years ago

I have large dataset of 1 class: drone, labeled in coco json format, which details can i modify to train your model in my dataset? Thank you!

WongKinYiu commented 3 years ago

You could try this: https://github.com/WongKinYiu/PyTorch_YOLOv4/issues/334