search

gjy3035 / GCC-SFCN

This is the official code of spatial FCN in the paper Learning from Synthetic Data for Crowd Counting in the Wild [CVPR2019].
https://gjy3035.github.io/GCC-CL/
MIT License
160 stars 39 forks source link

When I want to train with my own data set, I get the following error #19

Open Yellow-Champagne opened 5 years ago

Yellow-Champagne commented 5 years ago

When I want to train with my own data set, I get the following error D:\Program Files\Anaconda3\envs\lei\envs\py36-pytorch\python.exe" D:/zgy/GCC/GCC-SFCN-master/train.py Traceback (most recent call last): File “D:/zgy/GCC/GCC-SFCN-master/train.py”, line 250, in main() File “D:/zgy/GCC/GCC-SFCN-master/train.py”, line 100, in main i_tb = train(train_loader, net, optimizer, epoch, i_tb) File “D:/zgy/GCC/GCC-SFCN-master/train.py”, line 120, in train for i, data in enumerate(train_loader, 0): File “D:\Program Files\Anaconda3\envs\lei\envs\py36-pytorch\lib\site-packages\torch\utils\data\dataloader.py”, line 286, in next return self._process_next_batch(batch) File “D:\Program Files\Anaconda3\envs\lei\envs\py36-pytorch\lib\site-packages\torch\utils\data\dataloader.py”, line 307, in _process_next_batch raise batch.exc_type(batch.exc_msg) AssertionError: Traceback (most recent call last): File “D:\Program Files\Anaconda3\envs\lei\envs\py36-pytorch\lib\site-packages\torch\utils\data\dataloader.py”, line 57, in _worker_loop samples = collate_fn([dataset[i] for i in batch_indices]) File “D:\Program Files\Anaconda3\envs\lei\envs\py36-pytorch\lib\site-packages\torch\utils\data\dataloader.py”, line 57, in samples = collate_fn([dataset[i] for i in batch_indices]) File “D:\zgy\GCC\GCC-SFCN-master\datasets\UCF_QNRF.py”, line 40, in getitem img, den = self.main_transform(img,den) File “D:\zgy\GCC\GCC-SFCN-master\misc\transforms.py”, line 16, in call img, mask = t(img, mask) File “D:\zgy\GCC\GCC-SFCN-master\misc\transforms.py”, line 50, in call assert img.size == mask.size AssertionError