wangyuxin87 / ContourNet

A PyTorch implementation of "ContourNet: Taking a Further Step toward Accurate Arbitrary-shaped Scene Text Detection" (CVPR2020)
Other
227 stars 46 forks source link

CTW1500数据训练问题 #30

Open S130111 opened 3 years ago

S130111 commented 3 years ago

Traceback (most recent call last): File "tools/train_net.py", line 174, in main() File "tools/train_net.py", line 167, in main model = train(cfg, args.local_rank, args.distributed) File "tools/train_net.py", line 73, in train arguments, File "/home/root1/SHI/ContourNet/maskrcnn_benchmark/engine/trainer.py", line 57, in dotrain for iteration, (images, targets, ) in enumerate(data_loader, start_iter): File "/home/software/anaconda2019/envs/ContourNet/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 819, in next return self._process_data(data) File "/home/software/anaconda2019/envs/ContourNet/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 846, in _process_data data.reraise() File "/home/software/anaconda2019/envs/ContourNet/lib/python3.7/site-packages/torch/_utils.py", line 385, in reraise raise self.exc_type(msg) AttributeError: Caught AttributeError in DataLoader worker process 0. Original Traceback (most recent call last): File "/home/software/anaconda2019/envs/ContourNet/lib/python3.7/site-packages/torch/utils/data/_utils/worker.py", line 178, in _worker_loop data = fetcher.fetch(index) File "/home/software/anaconda2019/envs/ContourNet/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetch data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/software/anaconda2019/envs/ContourNet/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 44, in data = [self.dataset[idx] for idx in possibly_batched_index] File "/home/root1/SHI/ContourNet/maskrcnn_benchmark/data/datasets/word_dataset.py", line 64, in getitem img, anno = super(WordDataset, self).getitem(idx) File "/home/software/anaconda2019/envs/ContourNet/lib/python3.7/site-packages/torchvision/datasets/coco.py", line 118, in getitem img, target = self.transforms(img, target) File "/home/root1/SHI/ContourNet/maskrcnn_benchmark/data/transforms/transforms.py", line 26, in call image, target = t(image, target) File "/home/root1/SHI/ContourNet/maskrcnn_benchmark/data/transforms/transforms.py", line 143, in call crop_region, keep_target = self.adjust_crop_region(crop_region, target0) File "/home/root1/SHI/ContourNet/maskrcnn_benchmark/data/transforms/transforms.py", line 193, in adjust_crop_region inter_area = self.intersect_area(crop_region, target.bbox) AttributeError: 'dict' object has no attribute 'bbox'

求帮助