zhanghang1989 / zhanghang1989/PyTorch-Encoding
how can i train fcn on voc
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 2k
- Forks
- 448
- PR merge metrics
- No merged PRs in 30d
Description
i use pascal_aug as train voc12 as val, and batch size is None mean auto ,but i got
RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 500 and 375 in dimension 2 at /storage2/oujie/DFN/PyTorch
Encoding/pytorch/aten/src/TH/generic/THTensorMoreMath.cpp:1348
I think this is because image of voc have different size ,but i think this code use crop from image so should not have those kind of problem.
i also try batch=1
raise TypeError((error_msg.format(type(batch[0]))))
TypeError: batch must contain tensors, numbers, dicts or lists; found <class 'PIL.Image.Image'>
is there anyone know this ,please help me, thank you!!!
Contributor guide
No contributing guide indexed for this repository
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by reproducing the reported training setup with Pascal VOC data, including auto batch sizing and batch size 1, and inspect the full traceback. Compare the image and target shapes entering the batch, then confirm that the training run handles variable image sizes without the reported tensor-size or PIL.Image errors.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- computer-vision, machine-learning
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100