ZYK100 / LLCM

[CVPR 2023] Diverse Embedding Expansion Network and Low-Light Cross-Modality Benchmark for Visible-Infrared Person Re-identification
https://github.com/ZYK100/LLCM/blob/main/Agreement/LLCM%20DATASET%20RELEASE%20AGREEMENT.pdf
107 stars 12 forks source link

image size is not correct? #8

Closed zhaoxinshawn closed 1 year ago

zhaoxinshawn commented 1 year ago

Does the dataset need to be pre-processed when the image size is not correct?

Original Traceback (most recent call last):
  File "/home/zhaoxin/anaconda3/envs/python39/lib/python3.9/site-packages/torch/utils/data/_utils/worker.py", line 308, in _worker_loop
    data = fetcher.fetch(index)
  File "/home/zhaoxin/anaconda3/envs/python39/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 51, in fetch
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/home/zhaoxin/anaconda3/envs/python39/lib/python3.9/site-packages/torch/utils/data/_utils/fetch.py", line 51, in <listcomp>
    data = [self.dataset[idx] for idx in possibly_batched_index]
  File "/home/zhaoxin/projects/LLCM-main/DEEN/data_loader.py", line 28, in __getitem__
    img1 = self.transform(img1)
  File "/home/zhaoxin/anaconda3/envs/python39/lib/python3.9/site-packages/torchvision/transforms/transforms.py", line 95, in __call__
    img = t(img)
  File "/home/zhaoxin/anaconda3/envs/python39/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
    return forward_call(*args, **kwargs)
  File "/home/zhaoxin/anaconda3/envs/python39/lib/python3.9/site-packages/torchvision/transforms/transforms.py", line 688, in forward
    i, j, h, w = self.get_params(img, self.size)
  File "/home/zhaoxin/anaconda3/envs/python39/lib/python3.9/site-packages/torchvision/transforms/transforms.py", line 647, in get_params
    raise ValueError(f"Required crop size {(th, tw)} is larger than input image size {(h, w)}")
ValueError: Required crop size (384, 144) is larger than input image size (276, 148)
ZYK100 commented 1 year ago

For SYSU-MM01 dataset, you should run python pre_process_sysu.py to pepare the dataset. The images are resized into (384, 144) for our DEEN.