欢迎光临
我们一直在努力

maskrcnnselected_

项目场景:

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

赞(0)
【声明】:本博客不参与任何交易,也非中介,仅记录个人感兴趣的主机测评结果和优惠活动,内容均不作直接、间接、法定、约定的保证。访问本博客请务必遵守有关互联网的相关法律、规定与规则。一旦您访问本博客,即表示您已经知晓并接受了此声明通告。