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

问题描述：

训练时候报错

selected_polygons.append(self.polygons[i]) IndexError: list index out of range

原因分析：

我的pytorch版本是1.1.0。以下两个文件中

maskrcnn_benchmark/structures/segmentation_mask.py

maskrcnn_benchmark/modeling/balanced_positive_negative_sampler.py
找到了torch.bool，pytorch1.1.0中识别不了torch.bool。

解决方案：

将两个文件中torch.bool改为torch.uint8

参考链接：

https://blog.youkuaiyun.com/xddwz/article/details/113254615