标签:ROI img roi Python cv2 OpenCV pts 255
原文链接:https://blog.csdn.net/weierqiuba/article/details/70011375 原OpenCV-Python 选择ROI
2019年06月01日 18:07:36 SongpingWang 阅读数 454更多 分类专栏: OpenCV 计算机视觉OpenCV-Python选择ROI
1. 矩形ROI,直接使用切片,简单粗暴
import cv2
src = cv2.imread(“D:\python_script\ffff\qipan.png”)
roi = src[20:150, 50:300 ]
cv2.imshow(‘roi’,roi)
cv2.waitKey(0)
cv2.destroyAllWindows()
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
2. 使用OpenCV中的一些函数获取ROI区域,这个用的最多
import numpy as np
from matplotlib import pyplot as plt
import cv2
src = cv2.imread(“D:\python_script\ffff\qipan.png”)
ROI = np.zeros(src.shape, np.uint8) #感兴趣区域ROI
proimage0 = src.copy() #复制原图
“”“提取轮廓”""
proimage1=cv2.cvtColor(proimage0,cv2.COLOR_BGR2GRAY) #转换成灰度图
proimage2=cv2.adaptiveThreshold(proimage1,255,cv2.ADAPTIVE_THRESH_GAUSSIAN_C,cv2.THRESH_BINARY_INV,7,7)
proimage3,contours,hierarchy=cv2.findContours(proimage2,cv2.RETR_CCOMP,cv2.CHAIN_APPROX_NONE) #提取所有的轮廓
“”“ROI提取”""
cv2.drawContours(ROI, contours, 1,(255,255,255),-1) #ROI区域填充白色,轮廓ID1
ROI=cv2.cvtColor(ROI,cv2.COLOR_BGR2GRAY) #转换成灰度图
ROI=cv2.adaptiveThreshold(ROI,255,cv2.ADAPTIVE_THRESH_GAUSSIAN_C,cv2.THRESH_BINARY_INV,7,7) #自适应阈值化
imgroi= cv2.bitwise_and(ROI,proimage3) #图像交运算 ,获取的是原图处理——提取轮廓后的ROI
titles = [‘Original Image’, ‘proimage1’,‘proimage2’, ‘proimage3’]
images = [proimage0, proimage1, proimage2, proimage3]
for i in range(4):
plt.subplot(2,2,i+1),plt.imshow(images[i],‘gray’)
plt.title(titles[i])
plt.xticks([])
plt.yticks([])
plt.show()
cv2.imshow(‘roi’,roi)
cv2.imshow(‘imgroi’,imgroi)
cv2.waitKey(0)
cv2.destroyAllWindows()
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
- 30
- 31
- 32
- 33
- 34
- 35
- 36
3. 使用鼠标获取ROI区域
OpenCV-python自带 API 可直接选择矩形区域作为ROI,该API位于目标追踪模块,主要是cv2.selectROI()函数
import cv2
img = cv2.imread(“D:\python_script\ffff\qipan.png”)
roi = cv2.selectROI(windowName=“roi”, img=img, showCrosshair=True, fromCenter=False)
x, y, w, h = roi
cv2.rectangle(img=img, pt1=(x, y), pt2=(x + w, y + h), color=(0, 0, 255), thickness=2)
cv2.imshow(“roi”, img)
cv2.waitKey(0)
cv2.destroyAllWindows()
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
4. 鼠标获取多边形ROI
多边形ROI,主要利用鼠标交互进行绘制:
- 单击左键,选择多边形的点;
- 单击右键,删除最近一次选择的点;
- 单击中键,确定ROI区域并可视化。
- 按”S“键,将多边形ROI区域的点保存到本地”config.pkl"文件中
import cv2
import numpy as np
import joblib
pts = [] # 用于存放点
# 统一的:mouse callback function
def draw_roi(event, x, y, flags, param):
img2 = img.copy()
<span class="token keyword">if</span> event <span class="token operator">==</span> cv2<span class="token punctuation">.</span>EVENT_LBUTTONDOWN<span class="token punctuation">:</span> <span class="token comment"># 左键点击,选择点</span>
pts<span class="token punctuation">.</span>append<span class="token punctuation">(</span><span class="token punctuation">(</span>x<span class="token punctuation">,</span> y<span class="token punctuation">)</span><span class="token punctuation">)</span>
<span class="token keyword">if</span> event <span class="token operator">==</span> cv2<span class="token punctuation">.</span>EVENT_RBUTTONDOWN<span class="token punctuation">:</span> <span class="token comment"># 右键点击,取消最近一次选择的点</span>
pts<span class="token punctuation">.</span>pop<span class="token punctuation">(</span><span class="token punctuation">)</span>
<span class="token keyword">if</span> event <span class="token operator">==</span> cv2<span class="token punctuation">.</span>EVENT_MBUTTONDOWN<span class="token punctuation">:</span> <span class="token comment"># 中键绘制轮廓</span>
mask <span class="token operator">=</span> np<span class="token punctuation">.</span>zeros<span class="token punctuation">(</span>img<span class="token punctuation">.</span>shape<span class="token punctuation">,</span> np<span class="token punctuation">.</span>uint8<span class="token punctuation">)</span>
points <span class="token operator">=</span> np<span class="token punctuation">.</span>array<span class="token punctuation">(</span>pts<span class="token punctuation">,</span> np<span class="token punctuation">.</span>int32<span class="token punctuation">)</span>
points <span class="token operator">=</span> points<span class="token punctuation">.</span>reshape<span class="token punctuation">(</span><span class="token punctuation">(</span><span class="token operator">-</span><span class="token number">1</span><span class="token punctuation">,</span> <span class="token number">1</span><span class="token punctuation">,</span> <span class="token number">2</span><span class="token punctuation">)</span><span class="token punctuation">)</span>
<span class="token comment"># 画多边形</span>
mask <span class="token operator">=</span> cv2<span class="token punctuation">.</span>polylines<span class="token punctuation">(</span>mask<span class="token punctuation">,</span> <span class="token punctuation">[</span>points<span class="token punctuation">]</span><span class="token punctuation">,</span> <span class="token boolean">True</span><span class="token punctuation">,</span> <span class="token punctuation">(</span><span class="token number">255</span><span class="token punctuation">,</span> <span class="token number">255</span><span class="token punctuation">,</span> <span class="token number">255</span><span class="token punctuation">)</span><span class="token punctuation">,</span> <span class="token number">2</span><span class="token punctuation">)</span>
mask2 <span class="token operator">=</span> cv2<span class="token punctuation">.</span>fillPoly<span class="token punctuation">(</span>mask<span class="token punctuation">.</span>copy<span class="token punctuation">(</span><span class="token punctuation">)</span><span class="token punctuation">,</span> <span class="token punctuation">[</span>points<span class="token punctuation">]</span><span class="token punctuation">,</span> <span class="token punctuation">(</span><span class="token number">255</span><span class="token punctuation">,</span> <span class="token number">255</span><span class="token punctuation">,</span> <span class="token number">255</span><span class="token punctuation">)</span><span class="token punctuation">)</span> <span class="token comment"># 用于求 ROI</span>
mask3 <span class="token operator">=</span> cv2<span class="token punctuation">.</span>fillPoly<span class="token punctuation">(</span>mask<span class="token punctuation">.</span>copy<span class="token punctuation">(</span><span class="token punctuation">)</span><span class="token punctuation">,</span> <span class="token punctuation">[</span>points<span class="token punctuation">]</span><span class="token punctuation">,</span> <span class="token punctuation">(</span><span class="token number">0</span><span class="token punctuation">,</span> <span class="token number">255</span><span class="token punctuation">,</span> <span class="token number">0</span><span class="token punctuation">)</span><span class="token punctuation">)</span> <span class="token comment"># 用于 显示在桌面的图像</span>
show_image <span class="token operator">=</span> cv2<span class="token punctuation">.</span>addWeighted<span class="token punctuation">(</span>src1<span class="token operator">=</span>img<span class="token punctuation">,</span> alpha<span class="token operator">=</span><span class="token number">0.8</span><span class="token punctuation">,</span> src2<span class="token operator">=</span>mask3<span class="token punctuation">,</span> beta<span class="token operator">=</span><span class="token number">0.2</span><span class="token punctuation">,</span> gamma<span class="token operator">=</span><span class="token number">0</span><span class="token punctuation">)</span>
cv2<span class="token punctuation">.</span>imshow<span class="token punctuation">(</span><span class="token string">"mask"</span><span class="token punctuation">,</span> mask2<span class="token punctuation">)</span>
cv2<span class="token punctuation">.</span>imshow<span class="token punctuation">(</span><span class="token string">"show_img"</span><span class="token punctuation">,</span> show_image<span class="token punctuation">)</span>
ROI <span class="token operator">=</span> cv2<span class="token punctuation">.</span>bitwise_and<span class="token punctuation">(</span>mask2<span class="token punctuation">,</span> img<span class="token punctuation">)</span>
cv2<span class="token punctuation">.</span>imshow<span class="token punctuation">(</span><span class="token string">"ROI"</span><span class="token punctuation">,</span> ROI<span class="token punctuation">)</span>
cv2<span class="token punctuation">.</span>waitKey<span class="token punctuation">(</span><span class="token number">0</span><span class="token punctuation">)</span>
<span class="token keyword">if</span> <span class="token builtin">len</span><span class="token punctuation">(</span>pts<span class="token punctuation">)</span> <span class="token operator">></span> <span class="token number">0</span><span class="token punctuation">:</span>
<span class="token comment"># 将pts中的最后一点画出来</span>
cv2<span class="token punctuation">.</span>circle<span class="token punctuation">(</span>img2<span class="token punctuation">,</span> pts<span class="token punctuation">[</span><span class="token operator">-</span><span class="token number">1</span><span class="token punctuation">]</span><span class="token punctuation">,</span> <span class="token number">3</span><span class="token punctuation">,</span> <span class="token punctuation">(</span><span class="token number">0</span><span class="token punctuation">,</span> <span class="token number">0</span><span class="token punctuation">,</span> <span class="token number">255</span><span class="token punctuation">)</span><span class="token punctuation">,</span> <span class="token operator">-</span><span class="token number">1</span><span class="token punctuation">)</span>
<span class="token keyword">if</span> <span class="token builtin">len</span><span class="token punctuation">(</span>pts<span class="token punctuation">)</span> <span class="token operator">></span> <span class="token number">1</span><span class="token punctuation">:</span>
<span class="token comment"># 画线</span>
<span class="token keyword">for</span> i <span class="token keyword">in</span> <span class="token builtin">range</span><span class="token punctuation">(</span><span class="token builtin">len</span><span class="token punctuation">(</span>pts<span class="token punctuation">)</span> <span class="token operator">-</span> <span class="token number">1</span><span class="token punctuation">)</span><span class="token punctuation">:</span>
cv2<span class="token punctuation">.</span>circle<span class="token punctuation">(</span>img2<span class="token punctuation">,</span> pts<span class="token punctuation">[</span>i<span class="token punctuation">]</span><span class="token punctuation">,</span> <span class="token number">5</span><span class="token punctuation">,</span> <span class="token punctuation">(</span><span class="token number">0</span><span class="token punctuation">,</span> <span class="token number">0</span><span class="token punctuation">,</span> <span class="token number">255</span><span class="token punctuation">)</span><span class="token punctuation">,</span> <span class="token operator">-</span><span class="token number">1</span><span class="token punctuation">)</span> <span class="token comment"># x ,y 为鼠标点击地方的坐标</span>
cv2<span class="token punctuation">.</span>line<span class="token punctuation">(</span>img<span class="token operator">=</span>img2<span class="token punctuation">,</span> pt1<span class="token operator">=</span>pts<span class="token punctuation">[</span>i<span class="token punctuation">]</span><span class="token punctuation">,</span> pt2<span class="token operator">=</span>pts<span class="token punctuation">[</span>i <span class="token operator">+</span> <span class="token number">1</span><span class="token punctuation">]</span><span class="token punctuation">,</span> color<span class="token operator">=</span><span class="token punctuation">(</span><span class="token number">255</span><span class="token punctuation">,</span> <span class="token number">0</span><span class="token punctuation">,</span> <span class="token number">0</span><span class="token punctuation">)</span><span class="token punctuation">,</span> thickness<span class="token operator">=</span><span class="token number">2</span><span class="token punctuation">)</span>
cv2<span class="token punctuation">.</span>imshow<span class="token punctuation">(</span><span class="token string">'image'</span><span class="token punctuation">,</span> img2<span class="token punctuation">)</span>
# 创建图像与窗口并将窗口与回调函数绑定
img = cv2.imread(“D:\python_script\ffff\qipan.png”)
cv2.namedWindow(‘image’)
cv2.setMouseCallback(‘image’, draw_roi)
print("[INFO] 单击左键:选择点,单击右键:删除上一次选择的点,单击中键:确定ROI区域")
print("[INFO] 按‘S’确定选择区域并保存")
print("[INFO] 按 ESC 退出")
while True:
key = cv2.waitKey(1) & 0xFF
if key 27:
break
if key ord(“s”):
saved_data = {“ROI”: pts}
joblib.dump(value=saved_data, filename=“config.pkl”)
print("[INFO] ROI坐标已保存到本地.")
break
cv2.destroyAllWindows()
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
- 30
- 31
- 32
- 33
- 34
- 35
- 36
- 37
- 38
- 39
- 40
- 41
- 42
- 43
- 44
- 45
- 46
- 47
- 48
- 49
- 50
- 51
- 52
- 53
- 54
- 55
- 56
- 57
- 58
- 59
- 60
- 61
- 62
- 63
- 64
- 65
读取获取的ROI区域
def Load_Model(filepath):
img = cv2.imread("D:\\python_script\\ffff\\qipan.png")
model = joblib.load(filepath)
print(type(model))
print(model)
return model
Load_Model(‘config.pkl’)
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
</div>
<link href="https://csdnimg.cn/release/phoenix/mdeditor/markdown_views-e44c3c0e64.css" rel="stylesheet">
</div>
</article>
标签:ROI,img,roi,Python,cv2,OpenCV,pts,255 来源: https://blog.csdn.net/liuxiaohuidegushi/article/details/100865622
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。