selective_search.py

# Эта работа лицензирована по лицензии MIT.
# Copyright (c) 2013-2023 OpenMV LLC. All rights reserved.
# https://github.com/openmv/openmv/blob/master/LICENSE
#
# Пример выборочного поиска (Selective Search)

import csi
import time

csi0 = csi.CSI()
csi0.reset()  # Сброс и инициализация датчика.
csi0.pixformat(csi.RGB565)  # Установить формат пикселей RGB565 (или GRAYSCALE)
csi0.framesize(csi.QVGA)  # Установить размер кадра QVGA (320x240)
csi0.snapshot(time=2000)  # Подождать, пока настройки вступят в силу.
csi0.auto_gain(False)
csi0.auto_exposure(False, exposure_us=10000)

clock = time.clock()  # Создать объект clock для отслеживания FPS.


while True:
    clock.tick()  # Обновить счётчик FPS.
    img = csi0.snapshot()  # Сделать снимок и вернуть изображение.
    rois = img.selective_search(threshold=200, size=20, a1=0.5, a2=1.0, a3=1.0)
    for r in rois:
        img.draw_rectangle(r, color=(255, 0, 0))
        # from random import randint
        # img.draw_rectangle(r, color=(randint(100, 255), randint(100, 255), randint(100, 255)))
    print(clock.fps())