Hướng dẫn how do i create a multiple rectangle in opencv python? - làm cách nào để tạo nhiều hình chữ nhật trong opencv python?

Giới thiệu

Trong bài đăng này, chúng tôi sẽ học cách vẽ hình chữ nhật trên một hình ảnh, sử dụng Python và OpenCV.

OpenCV cung cấp cho chúng tôi một vài chức năng để vẽ hình dạng trên hình ảnh, như có thể thấy ở đây. Đây có thể là thú vị để làm nổi bật các khu vực quan tâm. Ví dụ, nếu chúng ta đang áp dụng một thuật toán để phát hiện các khuôn mặt, một cách đơn giản để báo hiệu chúng có thể vẽ một hình chữ nhật xung quanh chúng.

Hướng dẫn này đã được thử nghiệm trên Windows 8.1, với phiên bản 4.1.2 của OpenCV. Phiên bản Python được sử dụng là 3,7.2.

Vẽ hình chữ nhật

Dòng đầu tiên của mã của chúng tôi sẽ là nhập mô -đun CV2, vì vậy chúng tôi có quyền truy cập vào chức năng mà chúng tôi cần vẽ hình chữ nhật trên hình ảnh.cv2 module, so we have access to the function we need to draw rectangles on images.

Tiếp theo, chúng tôi sẽ chăm sóc đọc một hình ảnh từ hệ thống tệp. Chúng tôi làm như vậy với một cuộc gọi đến chức năng iMread. Là đầu vào, chúng tôi chuyển đường dẫn đến hình ảnh và, dưới dạng đầu ra, hàm trả về hình ảnh dưới dạng ndarray. Khi kiểm tra mã, đừng quên cập nhật đường dẫn lên hình ảnh trong hệ thống tệp của bạn.

img = cv2.imread('C:/Users/N/Desktop/monument.png')

Bây giờ chúng ta có hình ảnh của chúng ta trong bộ nhớ, chúng ta sẽ vẽ một số hình chữ nhật trên đó. Để vẽ một hình chữ nhật với OpenCV, chúng ta chỉ cần sử dụng hàm hình chữ nhật. Hàm này nhận được các đối số sau:

• Hình ảnh nơi vẽ hình chữ nhật. Chúng tôi sẽ vượt qua hình ảnh chúng tôi vừa đọc;
• Một tuple với tọa độ X và Y của một trong các đỉnh của hình chữ nhật. Để đơn giản hóa bản vẽ của hình dạng, việc luôn luôn nghĩ về đỉnh này là góc trên cùng bên trái của hình chữ nhật;
• Một tuple với tọa độ X và Y của đỉnh đối diện của hình chữ nhật, liên quan đến hình chữ nhật trước. Ngoài ra để đơn giản hóa, chúng ta có thể xem xét đây là góc dưới bên phải của hình chữ nhật;
• Một tuple với màu BGR của hình chữ nhật;
• Độ dày của các đường của hình chữ nhật. Đó là một tham số tùy chọn, khi không được chỉ định, mặc định là 1. Ngoài ra, các số âm sẽ dẫn đến một hình chữ nhật đầy được vẽ.

Điều quan trọng là phải xem xét rằng hàm hình chữ nhật làm thay đổi hình ảnh chúng ta truyền dưới dạng đầu vào. Điều này có nghĩa là, nếu chúng ta muốn giữ hình ảnh gốc không thay đổi, chúng ta nên thực hiện một bản sao của nó và vẽ hình chữ nhật trên bản sao đó. Để tìm hiểu cách thực hiện một bản sao của một hình ảnh trong bộ nhớ với OpenCV, vui lòng tham khảo bài đăng này.rectangle function alters the image we pass as input. This means that, if we want to keep the original image unaltered, we should do a copy of it and draw the rectangle on that copy. To learn how to perform a copy of an image in memory with OpenCV, please consult this post.

Bây giờ chúng tôi đã phân tích tất cả các đối số của hàm hình chữ nhật của mô -đun CV2, chúng tôi sẽ vẽ hình dạng đầu tiên. Chúng ta sẽ vẽ nó giữa (x = 10, y = 10) và (x = 100, y = 100), màu xanh lá cây. Ngoài ra, để minh họa rằng độ dày là tùy chọn, chúng tôi đã giành được giá trị này.rectangle function of the cv2 module, we will draw the first shape. We will draw it between (x=10, y=10) and (x=100, y=100), in green color. Also, to illustrate that the thickness is optional, we won’t set this value.

cv2.rectangle(img, (10, 10), (100, 100), (0, 255, 0))

Chúng ta sẽ vẽ một hình chữ nhật thứ hai, lần này là giữa (x = 120, y = 120) và (x = 150, y = 150). Màu sắc sẽ có màu xanh lần này và chúng tôi sẽ đặt độ dày của các đường của hình chữ nhật thành 5 pixel.

cv2.rectangle(img, (120, 120), (150, 150), (255, 0, 0), 5)

Hình chữ nhật cuối cùng của chúng tôi sẽ nằm giữa các tọa độ (x = 200, y = 200) và (x = 300, y = 400) và được vẽ màu đỏ. Chúng tôi cũng sẽ đặt độ dày thành một số âm, vì vậy nó được vẽ dưới dạng hình dạng đầy.

cv2.rectangle(img, (200, 200), (300, 400), (0, 0, 255), -1)

Quan trọng: Các tọa độ mà tôi đã sử dụng để vẽ ba hình chữ nhật trước đó là đủ cho kích thước của hình ảnh tôi sử dụng. Tùy thuộc vào hình ảnh bạn sẽ sử dụng cho các bài kiểm tra của mình, bạn có thể cần sử dụng các tọa độ khác nhau cho các hình chữ nhật để phù hợp với bên trong.: The coordinates I’ve used to draw the previous three rectangles are adequate for the size of the image I’m using. Depending on the image you will be using for your tests, you may need to use different coordinates for the rectangles to fit inside.

Bây giờ chúng tôi đã vẽ xong hình ảnh của mình, chúng tôi sẽ hiển thị nó trong một cửa sổ OpenCV. Điều này được thực hiện với một cuộc gọi đến chức năng Imshow, chuyển sang đầu tiên nhập tên của cửa sổ và là hình ảnh thứ hai của chúng tôi.imshow function, passing as first input the name of the window and as second our image.

Để hoàn thiện, chúng tôi sẽ đợi người dùng nhấn bất kỳ phím nào trên bàn phím và, khi điều đó xảy ra, phá hủy cửa sổ và kết thúc chương trình.

cv2.waitKey(0)
cv2.destroyAllWindows()

Mã Python hoàn chỉnh có thể được nhìn thấy dưới đây.

import cv2
 
img = cv2.imread('C:/Users/N/Desktop/monument.png')

cv2.rectangle(img, (10, 10), (100, 100), (0, 255, 0))
cv2.rectangle(img, (120, 120), (150, 150), (255, 0, 0), 5)
cv2.rectangle(img, (200, 200), (300, 400), (0, 0, 255), -1)


cv2.imshow('image', img)
 
cv2.waitKey(0)
cv2.destroyAllWindows()

Kiểm tra mã

Để kiểm tra tập lệnh Python trước đó, chỉ cần chạy nó trong một công cụ bạn chọn. Tôi sẽ sử dụng pycharm, một ide python.

Khi chạy mã, bạn sẽ nhận được kết quả tương tự như bản được minh họa trong Hình 1. Như có thể thấy, chúng tôi có ba hình chữ nhật trong hình ảnh. Cái đầu tiên được vẽ bằng màu xanh lá cây, với chiều rộng 1 pixel. Cái thứ hai được vẽ màu xanh lam và với các đường dày hơn. Cái thứ ba được vẽ màu đỏ và nó là một hình chữ nhật đầy. Ba được rút ra giữa các tọa độ mà chúng tôi đã chỉ định trong mã của chúng tôi, như mong đợi.

Bài đọc OpenCV đề xuất

• Cách vẽ vòng tròn
• Cách vẽ văn bản trên hình ảnh
• Phát hiện và đếm mặt