OpenCV - 커스텀 Cascade 만들기(CASCADE TRAINER GUI) 및 물체 탐지

아래 링크에서 Cascade Trainer GUI를 자신의 컴퓨터 환경에 맞는 버전으로 설치(64비트가 32비트 버전으로 설치하면 중간에 오류가 생김)

https://amin-ahmadi.com/cascade-trainer-gui/

Cascade Trainer GUI - Amin

 

디렉토리를 하나 만들어준 다음 안에다 n, p 폴더를 생성

n - negative: 찾고싶지 않는 이미지들을 여기에 저장

p - positive: 찾고자 하는 이미지들이 여기에 저장

 

그리고 옵션에서 Sample Width, Sample Height를 조정하는 부분이 있는데, 여기에는 찾고자 하는 이미지 파일들의 width, height를 나누기 10해준 사이즈를 넣어줌( 200x120 => 20x12 )

 

tip - 감지하고자 하는 물체는 최대한 다방면에서 찍는 것은 비추하고, 여러가지 다양한 배경이 나오도록 찍는 것을 추천

 

[물체 감지 스크립트]

import cv2

w_cascade = cv2.CascadeClassifier('cascade.xml')

cap = cv2.VideoCapture(0)

while cv2.waitKey(33) < 0:
    ret, img = cap.read()
    if ret:
        gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
        w = w_cascade.detectMultiScale(gray,
        scaleFactor = 1.3,
        minNeighbors = 10,
        minSize=(70, 70))

        for (x, y, w, h) in w:
            cv2.rectangle(img, (x, y), (x + w, y + h), (255, 255, 0), 2)
            font = cv2.FONT_HERSHEY_SIMPLEX
            cv2.putText(img, 'Detected', (x - w, y - h), font, 0.5, (11, 255, 255),  2, cv2.LINE_AA)

        cv2.imshow('img', img)

cap.release()
cv2.destroyAllWindows()

 

 

 

 

 

참고: 

https://www.youtube.com/watch?v=v_cwOq06g9E&t=47s&ab_channel=OMES 

https://www.youtube.com/watch?v=kLajDbq3eyI 

https://stackoverflow.com/questions/63609586/how-to-make-custom-any-object-cascade-xml-for-opencv-python

How to make custom 'any object' Cascade (.xml) for opencv-python?