python 基于detectron或mask_rcnn的mask遮罩区域进行图片截取
2024-09-05 07:16:49
基于示例infer_simple.py
修改165行vis_utils.vis_one_image
为vis_utils.vis_one_image_opencv
在detectron.utils.vis
中新增def cut_mask(img, mask)
def cut_mask(img, mask):
mask = mask*255
# 灰度图片‘cv2.cvtColor(mask,cv2.COLOR_BAYER_GB2GRAY)’转成三通道
mask = cv2.cvtColor(cv2.cvtColor(mask,cv2.COLOR_BAYER_GB2GRAY),cv2.COLOR_GRAY2BGR)
cv2.imshow('mask',mask)
ROI = cv2.bitwise_and(mask, img)
cv2.imshow('roi', ROI)
return ROI
在vis_one_image_opencv
方法中
注释
im = vis_mask(im, masks[..., i], color_mask)
改为调用
im = cut_mask(im, masks[..., i])
此时遮罩区域截图已经完成
查看上方的vis_bbox
方法可以获取识别框的位置与大小,这样就可以截取指定位置的遮罩区域。
补充:基于mask_rcnn注释152行masked_image = apply_mask(masked_image, mask, color)
改为上述调用。
最新文章
- Android Studio 快捷键
- POJ 1273 Drainage Ditches题解——S.B.S.
- log4g
- PHP测试用例文档
- UVA 558 判定负环,spfa模板题
- [CareerCup] 18.3 Randomly Generate Integers 随机生成数字
- xss-跨站脚本攻击-后台传给前端的html标签安全显示
- Linux&;shell之高级Shell脚本编程-创建函数
- iOS 深拷贝和浅拷贝
- 257. Binary Tree Paths
- 一个用 Cumulative Penalty 培训 L1 正规 Log-linear 型号随机梯度下降
- php上传图片到server
- APMServ—我用过的最优秀的PHP集成环境工具
- LanSoEditor_advance1.8.0 视频编辑的高级版本
- python3之shutil高级文件操作
- Android实现图片的压缩、旋转工具类
- python设计模式第五天【单例模式】
- 本博客已不再更新,新文章将发布在我的个人博客:https://www.tapme.top
- eclipse 远程调试程序
- python-day41--数据库---数据类型