如何解决如何自定义掩码 r-cnn 的 display_instances() 函数的输出
如何使用 display_instances() 函数只绘制没有边界框的蒙版,反之亦然?
visualize.display_instances(image=image,Boxes=r['rois'],masks=r['masks'],class_ids=r['class_ids'],class_names=class_names)
解决方法
将visualize.py 更新到以下版本,其中可以选择显示掩码或bbox。 updated visualize.py here
但是标签部分需要修改如下:
# Label
if not captions:
class_id = class_ids[i]
score = scores[i] if scores is not None else None
label = class_names[class_id]
caption = "{} {:.3f}".format(label,score) if score else label
ax.text(x1,y1 + 8,caption,color="none",size=11,backgroundcolor="none") # to remove captions change to: color= "none"
else:
caption = captions[i]
ax.text(x1,color='w',backgroundcolor="none") # to remove captions change to: color= "none"
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。