image size in args is not working

WongKinYiu / PyTorch_YOLOv4

PyTorch implementation of YOLOv4

1.89k stars 586 forks source link

image size in args is not working #228

Open Zxl19990529 opened 3 years ago

Zxl19990529 commented 3 years ago

Hi, I am running this master repo with yolov4.cfg. I find the image size option in train.py is not introduced to Darknet when creating the backbone module. The img_size is still using the default set 416*416. Is that correct?

https://github.com/WongKinYiu/PyTorch_YOLOv4/blob/a3e3a41d4447ce685793226890e8e88273f19ae9/models/models.py#L277

WongKinYiu commented 3 years ago

yes, you are correct.

haoxue1215 commented 3 years ago

yes, you are correct. 那如果我想更改图片尺寸，应该如何更改呢