OpenCV — это библиотека для программирования, предназначенная для компьютерного зрения в реальном времени. OpenCV-Python — это основная библиотека Python, специально разработанная для решения задач компьютерного зрения.
Предположим, вы работаете над проектами машинного обучения, часто сталкиваетесь со сценарием, когда вам нужно идентифицировать конкретный объект с изображения. В этом случае вам нужно нарисовать прямоугольник вокруг определенного объекта, чтобы отличить его от других объектов изображения.
Чтобы отделить этот объект, вам нужно нарисовать вокруг него прямоугольник, и тут на помощь приходит библиотека OpenCV-Python.
Что такое функция cv2.rectangle() в Python?
Чтобы нарисовать прямоугольник с помощью OpenCV в Python, используйте функцию cv2.rectangle(). Метод cv2 Rectangle() используется для рисования прямоугольника на любом изображении.
Синтаксис
|
1 |
cv2.rectangle(image, start_point, end_point, color, thickness) |
Параметры
- image: это фактическое изображение, на котором должен быть нарисован прямоугольник.
- start_point: start_point — это начальные координаты прямоугольника. Эти координаты представлены в виде кортежей из двух значений (значение координаты X, значение координаты Y).
- end_point: end_point — это конечные координаты прямоугольника. Эти координаты представлены в виде кортежей из двух значений (значение координаты X, значение координаты Y).
- color: этот параметр определяет границу прямоугольника. Для BGR мы передаем кортеж. например:(255, 0, 0) для синего цвета.
- thickness: этот параметр определяет толщину границы прямоугольника в пикселях. Толщина -1 пиксель заполнит прямоугольник указанным цветом.
Возвращаемое значение
Метод cv2.rectangle() возвращает изображение.
Установка Python OpenCV
Для работы с OpenCV в Python нам необходимо установить модуль opencv-python.
Введите следующую команду.
|
1 2 3 4 5 |
python3 -m pip install opencv-python # OR pip install opencv-python |
Чтобы использовать opencv-python в нашем проекте, мы должны импортировать модуль cv2 в файл.
|
1 |
import cv2 |
Пример метода cv2.rectangle()
Напишите следующий код внутри метода app.py.
|
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 |
# Importing cv2 import cv2 # Path to the local image path = 'cropped_image.jpg' # Reading an image in default mode image = cv2.imread(path) # Window name in which image is displayed window_name = 'Image' # Starting coordinate, here(100, 100) # Represents the top left corner of rectangle starting_point =(100, 100) # Ending coordinate, here(400, 400) # Represents the bottom right corner of rectangle ending_point =(400, 400) # Blue color in BGR color =(255, 0, 0) # Line thickness of 2 px thickness = 2 # Draw a rectangle with blue line borders of thickness of 2 px image = cv2.rectangle(image, starting_point, ending_point, color, thickness) # Displaying the image cv2.imshow(window_name, image) cv2.waitKey(0) # Closing all open windows cv2.destroyAllWindows() |
Вывод:
В этом примере сначала мы импортировали модуль cv2, а затем определили путь к входному изображению. Затем мы читаем изображение с помощью метода cv2.imread().
На следующем шаге мы определили имя окна — Image. Затем определил начальную точку и конечную точку. Эти координаты сообщают функции, когда начинать рисовать прямоугольник и где заканчивать. На следующем шаге мы определили цвет прямоугольника и толщину.
Для отображения изображения мы использовали метод cv2.imshow(). Чтобы сохранить окно изображения, мы использовали функцию waitKey(). Мы передали 0 методу waitKey(), что означает, что он останется открытым навсегда, пока мы не укажем обратное.
