Cat Detection

 

✅ CAT Detection

• Camera 인스턴스 생성
• 카메라 프리뷰
• 모델 가져오기
• 고양이 얼굴 감지!

from jetbot import Camera # 카메라 관련 라이브러리
from jetbot import bgr8_to_jpeg # bgr type -> jpeg 변환

camera = Camera.instance(width = 720, height = 720)

import ipywidgets.widgets as widgets #파이썬에서 위젯을 활용할 수 있는 라이브러리
from IPython.display import display #실제 이미지를 display할 수 있는 라이브러리

cat_detect = widgets.Image(width = 300, height = 300, format = 'jpeg') # 인식될 공간

display(cat_detect)

!curl -L -O https://github.com/opencv/opencv/raw/master/data/haarcascades/haarcascade_frontalcatface.xml

import cv2

cat_cascade = cv2.CascadeClassifier('haarcascade_frontalcatface.xml')

cat_x = cat_y = cat_w = cat_h = 0 # 좌표 초기화

while 1:
    frame = camera.value
    
    cv2.resize(frame, (300, 300))
    
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    
    cat = cat_cascade.detectMultiScale(gray)
    
    if len(cat) > 0:
        (cat_x, cat_y, cat_w, cat_h) = cat[0]
        
        # cv2.rectangle(frame,(cat_x,cat_y),(cat_x+cat_w,cat_y+cat_h),(255,0,0),3)
        cv2.circle(frame,(int(cat_x+(cat_w/2)),int(cat_y+(cat_h/2))),30,(255,0,0),3)
    
    cat_detect.value = bgr8_to_jpeg(frame)