首页 > 编程知识 正文

实例分割数据集制作,图像分割

时间:2023-05-04 06:51:04 阅读:12796 作者:2923

本文使用PointRend对Mask RCNN网络上遥感图像上的建筑物进行实例分割。 事实上,Facebook公式实现了PointRend,地址是https://github.com/Facebook research/detectron2/tree/master,但detectro N2项目另外,为了更清楚地了解PointRend在Mask Rcnn中使用的实现细节,我们将detectron2的PointRend部分移植到了torchvision版本的Mask Rcnn中。 上图是在此项目中培训的遥感影像中的建筑实例分割模型,加上PointRend,可以看到建筑分割的边界更加准确。

声明:本项目使用的torch==1.7.0,torchvision==0.8.1。 在实际测试中,发现不同版本的torchvison的检测部分的实现不匹配。 低版本的torchvison可能会出错,因为我根据0.8.1进行了修复

另一方面,MaskRCNN with PointRend试图在torchvison版本的Mask RCNN中使用PointRend时,首先需要对其进行一些更改。 (具体torchvision版本的MaskRcnn是

版权声明:该文观点仅代表作者本人。处理文章:请发送邮件至 三1五14八八95#扣扣.com 举报,一经查实,本站将立刻删除。