AttributeError: модуль 'cv2' не имеет атрибута 'selectROI' - PullRequest
1 голос
/ 19 июня 2019

У меня новая версия OpenCV 3.2 с contribs, ffmpeg и numpy. Однако, когда я пытаюсь использовать функцию selectROI, я получаю ошибку атрибута и не могу понять, почему !!!

Я пытался переустановить opencv и opencv-contrib, но это, похоже, ничего не меняет.

import numpy as np
import ffmpy
import cv2
import os

def main():
   ...
            r=0
            cap = cv2.VideoCapture(filename)
           ...
            while cap.grab():
               ...
                if (frame_count>=next_valid):
                    # Initialisation of supporting variables
                    flag, frame = cap.retrieve()
                    if (go_around==0):
                        # Select ROI
                        r = cv2.selectROI(frame)
                    # Cropping and Brightening
                    imCrop = im[int(r[1]):int(r[1]+r[3]), int(r[0]):int(r[0]+r[2])]
                   ...
main()

Я просто хотел бы выбрать рентабельность инвестиций и сохранить размеры!

1 Ответ

0 голосов
/ 19 июня 2019

enter image description here

Вы всегда можете сделать свой собственный селектор ROI

import cv2

class ExtractImageWidget(object):
    def __init__(self):
        self.original_image = cv2.imread('1.jpg')

        # Resize image, remove if you want raw image size
        self.original_image = cv2.resize(self.original_image, (640, 556))
        self.clone = self.original_image.copy()

        cv2.namedWindow('image')
        cv2.setMouseCallback('image', self.extract_coordinates)

        # Bounding box reference points and boolean if we are extracting coordinates
        self.image_coordinates = []
        self.extract = False

    def extract_coordinates(self, event, x, y, flags, parameters):
        # Record starting (x,y) coordinates on left mouse button click
        if event == cv2.EVENT_LBUTTONDOWN:
            self.image_coordinates = [(x,y)]
            self.extract = True

        # Record ending (x,y) coordintes on left mouse bottom release
        elif event == cv2.EVENT_LBUTTONUP:
            self.image_coordinates.append((x,y))
            self.extract = False
            print('top left: {}, bottom right: {}'.format(self.image_coordinates[0], self.image_coordinates[1]))

            # Draw rectangle around ROI
            cv2.rectangle(self.clone, self.image_coordinates[0], self.image_coordinates[1], (0,255,0), 2)
            cv2.imshow("image", self.clone) 

        # Clear drawing boxes on right mouse button click
        elif event == cv2.EVENT_RBUTTONDOWN:
            self.clone = self.original_image.copy()

    def show_image(self):
        return self.clone

if __name__ == '__main__':
    extract_image_widget = ExtractImageWidget()
    while True:
        cv2.imshow('image', extract_image_widget.show_image())
        key = cv2.waitKey(1)

        # Close program with keyboard 'q'
        if key == ord('q'):
            cv2.destroyAllWindows()
            exit(1)
...