Yolov3: dataloader

[MichaelCong]

Traceback (most recent call last):
File "train.py", line 308, in
multi_scale=opt.multi_scale,
File "train.py", line 168, in train
for i, (imgs, targets, _, _) in enumerate(dataloader):
File "/home/rencong/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 582, in __next__
return self._process_next_batch(batch)
File "/home/rencong/anaconda3/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 608, in _process_next_batch
raise batch.exc_type(batch.exc_msg)
RuntimeError: Traceback (most recent call last):
File "/home/rencong/anaconda3/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 99, in _worker_loop
samples = collate_fn([dataset[i] for i in batch_indices])
File "/home/rencong/yolov3-6/utils/datasets.py", line 338, in collate_fn
return torch.stack(img, 0), torch.cat(label, 0), path, hw
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 288 and 416 in dimension 2 at /pytorch/aten/src/TH/generic/THTensor.cpp:711

[glenn-jocher]

Hello, thank you for your interest in our work! This is an automated response. Please note that most technical problems are due to:

• Your changes to the default repository. If your issue is not reproducible in a fresh git clone version of this repository we can not debug it. Before going further run this code and ensure your issue persists:

sudo rm -rf yolov3  # remove exising repo
git clone https://github.com/ultralytics/yolov3 && cd yolov3 # git clone latest
python3 detect.py  # verify detection
python3 train.py  # verify training (a few batches only)
# CODE TO REPRODUCE YOUR ISSUE HERE

• Your custom data. If your issue is not reproducible with COCO data we can not debug it. Visit our Custom Training Tutorial for exact details on how to format your custom data. Examine train_batch0.jpg and test_batch0.jpg for a sanity check of training and testing data.
• Your environment. If your issue is not reproducible in a GCP Quickstart Guide VM we can not debug it. Ensure you meet the requirements specified in the README: Unix, MacOS, or Windows with Python >= 3.7, Pytorch >= 1.1, etc. You can also use our Google Colab Notebook to test your code in working environment.

If none of these apply to you, we suggest you close this issue and raise a new one using the Bug Report template, providing screenshots and minimum viable code to reproduce your issue. Thank you!

[edmondponycode]

@MichaelCong have you found a way to resolve this issue ? I have the same