When I run eval.py, it occurs
./datasets/total_text/
[INFO] [2020-11-13 11:21:48,157] Resuming from /data/dq/work/DB-master/models/totaltext_resnet18
[INFO] [2020-11-13 11:21:48,415] Resumed from /data/dq/work/DB-master/models/totaltext_resnet18
0%| | 0/300 [00:00<?, ?it/s]
Traceback (most recent call last):
File "eval.py", line 193, in
main()
File "eval.py", line 79, in main
Eval(experiment, experiment_args, cmd=args, verbose=args['verbose']).eval(args['visualize'])
File "eval.py", line 172, in eval
for i, batch in tqdm(enumerate(data_loader), total=len(data_loader)):
File "/home/dq/anaconda3/envs/db/lib/python3.8/site-packages/tqdm/std.py", line 1171, in iter
for obj in iterable:
File "/home/dq/anaconda3/envs/db/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 345, in next
data = self._next_data()
File "/home/dq/anaconda3/envs/db/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 856, in _next_data
return self._process_data(data)
File "/home/dq/anaconda3/envs/db/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 881, in _process_data
data.reraise()
File "/home/dq/anaconda3/envs/db/lib/python3.8/site-packages/torch/_utils.py", line 394, in reraise
raise self.exc_type(msg)
AttributeError: Caught AttributeError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/home/dq/anaconda3/envs/db/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 178, in _worker_loop
data = fetcher.fetch(index)
File "/home/dq/anaconda3/envs/db/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 "/home/dq/anaconda3/envs/db/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 "/data/dq/work/DB-master/data/image_dataset.py", line 84, in getitem
img = cv2.imread(image_path, cv2.IMREAD_COLOR).astype('float32')
AttributeError: 'NoneType' object has no attribute 'astype'
my environment is CUDA10.1, pytorch 1.4. Does anyone know how to solve it?
DB copied to clipboard
MhLiao
