Select and manipulate Region of interest.
A small python module to select a polygonal region of interest (ROI) in a given image that is stored as a Shape object. You can use this Shape object later to manipulate the polygon selected. You can also extract the inner content from an image, calculate the histogram of the created shape, calculate the center of the shape, rotate the shape around its center, or translate the shape.
import cv2 as cv
from polyroi import Shape
img = cv.imread('image.jpg')
shape = Shape.get_roi(img) #returns a Shape object
shape.draw_shape(img, color=(0, 255, 255), thickness=1)
while(1):
cv.imshow("Getting Started", img)
k = cv.waitKey(1) & 0xFF
if k == 27:
break
cv.destroyAllWindows()
pip install cv2
pip install numpy
pip install polyroi
Some time ago, I looked for an efficient tool to draw and manipulate polygons in a python environment. But I didn't find anything useful for my case. I did find some tools that can draw and extract a NumPy array, but as for the manipulation of shapes, I had to develop the logic myself. So I decided to create one. I was trying to implement the particle filter from Part-Based Lumbar Vertebrae Tracking in Videofluoroscopy Using Particle Filter. You can check the repository of how I did manage to work with this package.
img = cv.imread('image.jpg')
# returns a Shape object
shape = Shape.get_roi(img)
# Copy the shape
shape2 = Shape.copy(shape)
# Rotate the shape
shape2.rotate_around_center(np.pi/4)
# x translate the shape by 5
shape2.translate_x(5)
# y translate the shape by 5
shape2.translate_y(5)
# recalculate the center of the shape
shape2.centroid()
# translate the shape first point to (10, 15) along with the shape
shape2.translate_to(10, 15)
# x translate, y translate, and rotate around the center by np.pi / 12
shape2.update(5, 3, np.pi / 12)
# Drawing the shapes
shape.draw_shape(img, color=(0, 255, 255), thickness=1)
shape2.draw_shape(img, color=(0, 255, 0), thickness=1)
# return the bounding box points (upper left, bottom right)
p1, p2 = shape2.to_rectangle()
# plotting the image
while(1):
cv.imshow("Getting Started", img)
k = cv.waitKey(1) & 0xFF
if k == 27:
break
cv.destroyAllWindows()
- @skywolfmo - Idea & Initial work
See also the list of contributors who participated in this project.