Ошибка при попытке использовать размытие с Haarcascades - PullRequest
1 голос
/ 05 мая 2020

Я пытаюсь сделать проект самостоятельно, и я думал, что смогу это сделать, но эта проблема возникла и хорошо ... Я не знаю, что происходит. Цель этого проекта - размыть глаза на фотографиях или даже видео.

import matplotlib.pyplot as plt
import numpy as np
import cv2

people = cv2.imread('Computer-Vision-with-Python/DATA/people.jpg',0)
people2 = cv2.imread('Computer-Vision-with-Python/DATA/people2.jpg')

def display(img, cmap='gray'):
    fig=plt.figure(figsize=(12,10))
    ax = fig.add_subplot(111)
    ax.imshow(img,cmap='gray')

eye_cascade = cv2.CascadeClassifier('Computer-Vision-with-Python/DATA/haarcascades/haarcascade_eye.xml')

def detect_eye(img):
    face_img = img.copy()

    face_rects = eye_cascade.detectMultiScale(face_img,scaleFactor=1.2,minNeighbors=6)

    for (x,y,w,h) in face_rects:
        cv2.rectangle(face_img,(x,y),(x+w,y+h),(255,255,255),10)

    return face_img

def detect_and_blur_eye(img):

    eye_img = img.copy()
    roi = img.copy()

    eye_rects = eye_cascade.detectMultiScale(eye_img,scaleFactor=1.2, minNeighbors=6) 

    for (x,y,w,h) in eye_rects: 
        print (x,y,w,h)
        roi = roi[y:y+h,x:x+w]
        blurred_roi = cv2.medianBlur(roi,7)

        eye_img[y:y+h,x:x+w] = blurred_roi

    return eye_img

results = detect_and_blur_eye(people)

После этого я получаю эту ошибку:


TypeError                                 Traceback (most recent call last)
<ipython-input-259-f561e117d7f8> in <module>
----> 1 results = detect_and_blur_eye(people)

<ipython-input-258-2ab32e080f88> in detect_and_blur_eye(img)
     12         blurred_roi = cv2.medianBlur(roi,7)
     13 
---> 14         eye_img[y:y+h,x:x+w] = blurred_roi
     15 
     16     return eye_img

TypeError: int() argument must be a string, a bytes-like object or a number, not 'NoneType'

Я не знаю, что происходит! Пожалуйста, помогите мне в этой ситуации. Заранее спасибо.

EDIT

Мне удалось обнаружить кое-что, что может помочь мне решить проблему. Я изменил функцию, чтобы получить некоторые значения.

def detect_and_blur_eye(img):

    eye_img = img.copy()
    roi = img.copy()

    eye_rects = eye_cascade.detectMultiScale(eye_img,scaleFactor=1.2, minNeighbors=6) 

    for (x,y,w,h) in eye_rects: 
        print (type(eye_rects))
        print (x,y,w,h)
        roi = roi[y:y+h,x:x+w]
        print(roi.shape)
        #blurred_roi = cv2.medianBlur(roi,7)

        #eye_img[y:y+h,x:x+w] = blurred_roi

    return eye_img

results = detect_and_blur_eye(people)

И он возвращает мне это:

<class 'numpy.ndarray'>
1182 414 45 45
(45, 45)
<class 'numpy.ndarray'>
595 427 56 56
(0, 0)
<class 'numpy.ndarray'>
512 430 57 57
(0, 0)
<class 'numpy.ndarray'>
270 470 60 60
(0, 0)
<class 'numpy.ndarray'>
349 475 56 56
(0, 0)
<class 'numpy.ndarray'>
981 375 62 62
(0, 0)
<class 'numpy.ndarray'>
842 389 50 50
(0, 0)
<class 'numpy.ndarray'>
762 391 50 50
(0, 0)
<class 'numpy.ndarray'>
1072 390 54 54
(0, 0)
<class 'numpy.ndarray'>
1238 399 48 48
(0, 0)

Итак, по какой-то причине моя форма (0,0)

Ответы [ 2 ]

0 голосов
/ 05 мая 2020

Мне удалось обнаружить проблему. В этой функции мне нужно сбросить ROI, прежде чем я получу другое значение, а я этого не делал.

def detect_and_blur_eye(img):

    eye_img = img.copy()
    roi = img.copy()

    eye_rects = eye_cascade.detectMultiScale(eye_img,scaleFactor=1.2, minNeighbors=6,minSize=(5,5)) 

    for (x,y,w,h) in eye_rects:
        roi = img.copy()
        print (type(eye_rects))
        print (x,y,w,h)
        roi = roi[y:y+h,x:x+w]
        print(roi.shape)
        blurred_roi = cv2.medianBlur(roi,21)
        eye_img[y:y+h,x:x+w] = blurred_roi

    return eye_img

Это работает!

0 голосов
/ 05 мая 2020

Да, ваша проблема, как сказал @ jtlz2, связана с тем, что глаза не обнаруживаются на изображениях и, следовательно, не имеют никакого типа, который, когда вы пытаетесь назначить его, вызывает ошибку. Для этого могут быть два обходных пути: либо вы используете лучший классификатор, который заполняет глаза поиска, но затем, если вы хотите использовать его для видео, вам понадобится режим реального времени и не сможете использовать CNN. Альтернативой может быть dlib face key points. Помимо этого, если вы хотите продолжить только каскады хаара, вы можете сделать это:

try:
    print (x,y,w,h)
    roi = roi[y:y+h,x:x+w]
    blurred_roi = cv2.medianBlur(roi,7)

    eye_img[y:y+h,x:x+w] = blurred_roi
except:
    pass

Таким образом, если есть какие-либо ошибки, ваши глаза не будут размыты, но программа не будет генерировать ошибка.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...