项目场景:
maskrcnn训练问题报错:selected_polygons.append(self.polygons[i]) IndexError: list index out of range
问题描述:
训练时候报错
香港vps
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.csdn.net/xddwz/article/details/113254615
78975904