pytorch-animeGAN icon indicating copy to clipboard operation
pytorch-animeGAN copied to clipboard

Is this a problem with the image itself or with the code?

Open zwq11 opened this issue 2 years ago • 5 comments

Init models.. . Compute mean (R,G,B)from 1792 images100%|[ Mean(B,G, R)of Hayao are [-4.4346958-8.6659159713.10061177]Dataset: real 9 style 1792,smooth 1793 Epoch 0/100 O%| Traceback (most recent call last): File "train. py", line 247,in main(args) File "train. py",line 162,in mainfor img,*_ in bar: File "/root/miniconda3/envs/AGAN/lib/python3.8/site-packages/tqdm/ / tadm. py",line 919,in _iter for obj in iterable: File"/root/miniconda3/envs/AGAV/1ib/python3.8/site-packages/torch/utils/data/dataloader.py",line 517,in_nextdata = self._next_data() File "/root/minicandla3/envs/AGNV1ib/python3.8/site packages/torch/utils/data/dataloader. py",line 1199,in _next_datareturn self. _process_data(data) File "/rot/minicomda83/envs /ACGNV/lib/python3.8/site-packages/torch/utils/data/dataloader.py",line 1225,in _process_datadata.reraise() File "/root/miniconda3/envs/AGAV/1ib/python3.8/site packages/torch/_utils.py",line 429,in reraiseraise self.exc_type(msg) TypeError: Caught TypeError in DataLoader worker process 0.Original Traceback (most recent call last): File "/root/miniconda3/envs/AGNV1ib/python3.8/site-packages/torch/utils/data/l_utils/worker.py",line 202,in _worker_loopdata = fetcher.fetch(index) File"/root/miniconda3/envs/AGAV1ib/python3.8/site packages/torch/utils/data/_utils/fetch. py",line 44,in fetchdata = [self.dataset[idx] for idx in possibly_batched_index] File "/root/miniconda3/envs/AGAV/1ib/python3.8/site packages/torch/uti1s/data/_utils/fetch.py", line 44,in data = [self.dataset[idx] for idx in possibly_batched_index] File "/root/autodl-tmp/pytorch-animeGAN-master/dataset.py",line 67,in _getitemimage = self.load photo(index) File "/root/autodl-tmp/pytorch-animeGAN-master/dataset.py",line 79, in load photoimage = cv2.imread(fpath)[:,:,::-1] TypeError: 'NoneType’object is not subscriptable

zwq11 avatar Mar 01 '22 08:03 zwq11

TypeError: 'NoneType’object is not subscriptable

This error indicate that image is None, probably you want to check the image path

ptran1203 avatar Mar 01 '22 08:03 ptran1203

Already checked the path, can not adjust the problem, there is a better way to solve?

zwq11 avatar Mar 01 '22 09:03 zwq11

image = cv2.imread(fpath)
assert image is not None, f"{fpath}, {os.path.isfile(fpath)}"
image = image[:,:,::-1]

Use this code to make sure you load image correctly

ptran1203 avatar Mar 01 '22 09:03 ptran1203

Traceback (most recent call last): File "train.py", line 248, in main(args) File "train.py", line 163, in main for img, *_ in bar: File "/root/miniconda3/envs/AGAN/lib/python3.8/site-packages/tqdm/_tqdm.py", line 919, in iter for obj in iterable: File "/root/miniconda3/envs/AGAN/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 517, in next data = self._next_data() File "/root/miniconda3/envs/AGAN/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1199, in _next_data return self._process_data(data) File "/root/miniconda3/envs/AGAN/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1225, in _process_data data.reraise() File "/root/miniconda3/envs/AGAN/lib/python3.8/site-packages/torch/_utils.py", line 429, in reraise raise self.exc_type(msg) AssertionError: Caught AssertionError in DataLoader worker process 8. Original Traceback (most recent call last): File "/root/miniconda3/envs/AGAN/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 202, in _worker_loop data = fetcher.fetch(index) File "/root/miniconda3/envs/AGAN/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/root/miniconda3/envs/AGAN/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 44, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/root/autodl-tmp/pytorch-animeGAN-master/dataset.py", line 69, in getitem image = self.load_photo(index) File "/root/autodl-tmp/pytorch-animeGAN-master/dataset.py", line 83, in load_photo assert image is not None, f"{fpath}, {os.path.isfile(fpath)}" AssertionError: content/dataset/train_photo/.ipynb_checkpoints, False### The output looks like this, how do I load the image out?

zwq11 avatar Mar 01 '22 09:03 zwq11

Already solved, thank you very much

zwq11 avatar Mar 01 '22 10:03 zwq11