maskrcnn训练问题报错：selected_polygons.append(self.polygons[i]) IndexError: list index out of range

训练mask rcnn实例分割时报错：

Traceback (most recent call last):File "tools/train_net.py", line 203, in <module>main()File "tools/train_net.py", line 196, in mainmodel = train(cfg, args.local_rank, args.distributed)File "tools/train_net.py", line 84, in traincfg,File "/home/lyh/Domain-Adaptive-Faster-RCNN-PyTorch/maskrcnn_benchmark/engine/trainer.py", line 137, in do_da_trainfor iteration, ((source_images, source_targets, idx1), (target_images, target_targets, idx2)) in enumerate(zip(source_data_loader, target_data_loader), start_iter):File "/opt/conda/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 819, in __next__return self._process_data(data)File "/opt/conda/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 846, in _process_datadata.reraise()File "/opt/conda/lib/python3.6/site-packages/torch/_utils.py", line 385, in reraiseraise self.exc_type(msg)
IndexError: Caught IndexError in DataLoader worker process 0.
Original Traceback (most recent call last):File "/opt/conda/lib/python3.6/site-packages/torch/utils/data/_utils/worker.py", line 178, in _worker_loopdata = fetcher.fetch(index)File "/opt/conda/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 44, in fetchdata = [self.dataset[idx] for idx in possibly_batched_index]File "/opt/conda/lib/python3.6/site-packages/torch/utils/data/_utils/fetch.py", line 44, in <listcomp>data = [self.dataset[idx] for idx in possibly_batched_index]File "/home/lyh/Domain-Adaptive-Faster-RCNN-PyTorch/maskrcnn_benchmark/data/datasets/coco.py", line 95, in __getitem__target = target.clip_to_image(remove_empty=True)File "/home/lyh/Domain-Adaptive-Faster-RCNN-PyTorch/maskrcnn_benchmark/structures/bounding_box.py", line 223, in clip_to_imagereturn self[keep]File "/home/lyh/Domain-Adaptive-Faster-RCNN-PyTorch/maskrcnn_benchmark/structures/bounding_box.py", line 208, in __getitem__bbox.add_field(k, v[item])File "/home/lyh/Domain-Adaptive-Faster-RCNN-PyTorch/maskrcnn_benchmark/structures/segmentation_mask.py", line 203, in __getitem__selected_polygons.append(self.polygons[i])
IndexError: list index out of range

解决方法：

修改以下两个文件，将文件中的torch.uint8改为torch.bool即可

maskrcnn_benchmark/structures/segmentation_mask.py

maskrcnn_benchmark/modeling/balanced_positive_negative_sampler.py

maskrcnn训练问题报错：selected_polygons.append(self.polygons[i]) IndexError: list index out of range相关推荐

maskrcnn selected_polygons.append(self.polygons[i]) IndexError: list index out of range
项目场景: maskrcnn训练问题报错:selected_polygons.append(self.polygons[i]) IndexError: list index out of range ...
成功解决.append方法出现错误IndexError: list index out of range
成功解决.append方法出现错误IndexError: list index out of range 目录解决问题解决方法解决问题 .append方法出现错误IndexError: list ...
Tensorflow CIFAR-10训练例子报错解决
Tensorflow CIFAR-10训练例子报错解决大部分是Api版本问题: AttributeError: 'module' object has no attribute 'SummaryWr ...
使用pcb-tools解码Gerber文件报错IndexError: list index out of range
问题描述: 基于Github提供的示例代码编写: import gerber from gerber.render.cairo_backend import GerberCairoContext# R ...
Git报错：fatal: Unable to create ‘xx/.git/index.lock‘: File exis
一报错:fatal: Unable to create 'xx/.git/index.lock': File exis 如下: $ git commit fatal: Unable to creat ...
mmdetection训练出现：IndexError: list index out of range 错误
mmdetection训练出现:IndexError: list index out of range 错误文章目录: 1 问题分析 1.1 尝试解决错误:第一次 1.2 尝试解决错误:第二次 2 ...
Sampler 在数据下沉模式超时；不同Sampler策略，在非数据下沉模式下，模型训练失败报错Segmentation fault(core dumped)
环境版本 mindspore-ascend 1.7.1 背景我在使用自定义Sampler策略,对imagenet1K数据集进行采样,进行ResNet50训练. Sampler代码如下: class ...
姿态估计1-02：HR-Net(人体姿态估算)-官方模型训练测试-报错解决
以下链接是个人关于HR-Net(人体姿态估算)所有见解,如有错误欢迎大家指出,我会第一时间纠正.有兴趣的朋友可以加微信:17575010159 相互讨论技术.若是帮助到了你什么,一定要记得点赞!因为这 ...
使用GPU训练yolo_v2报错：RuntimeError: CUDA out of memory. Tried to allocate XXX MiB
报错内容: RuntimeError: CUDA out of memory. Tried to allocate 200.00 MiB (GPU 0; 8.00 GiB total capacity ...

maskrcnn训练问题报错：selected_polygons.append(self.polygons[i]) IndexError: list index out of range

maskrcnn训练问题报错：selected_polygons.append(self.polygons[i]) IndexError: list index out of range相关推荐

最新文章

热门文章