How to copy a image region using opencv in python?
Both cv.GetSubRect and ROI functions are available in Python, but in old import cv
mode or import cv2.cv
. ie use cv2.cv.GetSubRect()
or cv2.cv.SetImageROI
if you are familier with them.
On the other hand, it is simple to set ROI without these functions due to numpy integration in new cv2.
If (x1,y1) and (x2,y2) are the two opposite vertices of plate you obtained, then simply use function:
roi = gray[y1:y2, x1:x2]
that is your image ROI.
So choose whatever suit you.
Here's a visualization for cropping a ROI from an image
-------------------------------------------
| |
| (x1, y1) |
| ------------------------ |
| | | |
| | | |
| | ROI | |
| | | |
| | | |
| | | |
| ------------------------ |
| (x2, y2) |
| |
| |
| |
-------------------------------------------
Consider (0,0)
as the top-left corner of the image with left-to-right as the x-direction and top-to-bottom as the y-direction. If we have (x1,y1)
as the top-left and (x2,y2)
as the bottom-right vertex of a ROI, we can use Numpy slicing to crop the image with:
ROI = image[y1:y2, x1:x2]
But normally we will not have the bottom-right vertex. In typical cases, we will be iterating through contours where the rectangular ROI coordinates can be found with cv2.boundingRect()
. Additionally, if we wanted to save multiple ROIs, we could keep a counter
cnts = cv2.findContours(grayscale_image, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
cnts = cnts[0] if len(cnts) == 2 else cnts[1]
ROI_number = 0
for c in cnts:
x,y,w,h = cv2.boundingRect(c)
ROI = image[y:y+h, x:x+w]
cv2.imwrite('ROI_{}.png'.format(ROI_number), ROI)
ROI_number += 1
Since OpenCV v2.2, Numpy arrays are naively used to display images. This Numpy slicing method to extract the ROI may not work with older versions
Example: If you have few points, and want to copy region contains its
r = cv2.boundingRect(pts)
cv2.imwrite('roi.png', im[r[0]:r[0]+r[2], r[1]:r[1]+r[3]])