search

WongKinYiu / PyTorch_YOLOv4

PyTorch implementation of YOLOv4
1.88k stars 585 forks source link

what about yolov3？ #46

Open yunxiaozimeng opened 4 years ago

yunxiaozimeng commented 4 years ago

can this code be used with yolov3 and yolov3-tiny，if i have .cfg and .weights files?

WongKinYiu commented 4 years ago

yes, just change one line https://github.com/WongKinYiu/PyTorch_YOLOv4/blob/master/models.py#L102 to

if any(x in cfg for x in ['yolov4-tiny', 'yolov3', 'yolov3-tiny']):