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 to train with Image size 416 416? #232

Open Minseokkim-0124 opened 2 years ago

Minseokkim-0124 commented 2 years ago

I want to test with image size 416 but, because of max stride the img size stretch to 448. WARNING: --img-size 416 must be multiple of max stride 64, updating to 448 (In colab I can see these Warning words)

If I edit the code in the train.py # Image sizes gs = 52 #int(max(model.stride)) # grid size (max stride) imgsz, imgsz_test = [check_img_size(x, gs) for x in opt.img_size] # verify imgsz are gs-multiples If the error occurs what can I do?