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

[Yellow-Champagne]

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