[OpenCV] 실시간 영상 처리 : 기본 함수 정리

라즈베리파이의 카메라 영상을 실시간으로 인공지능 모델을 돌려서 감지를 해야하기 때문에 OpenCV 코드를 살펴보았다!

 

💡 영상을 인공지능 모델에 돌리는 법

1. 이미지 사용한 딥러닝 모델 생성
2. 동영상 프레임 → 이미지 (opencv 사용)
3. 2번의 이미지를 1번의 모델에 넣기 → 확률
4. 확률 평균 → 판별

---

OpenCV

• cap = cv2.VideoCapture(path or 카메라장치번호)

cap = cv2.VideoCapture("영상 경로" or 0: 카메라장치번호)

첫 프레임 읽어 cap 객체에 저장

• cap.isOpen()

if cap.isOpened():

cap이 지정한 프레임으로 제대로 초기화되었는지 확인

• cap.read()

ret, img = cap.read()  # 무한반복

연속으로 프레임 읽기 (영상 정보 읽기)

ret : true or false / img : 프레임 이미지 or None

• cap.get()

cap.get() # 영상정보 읽어오기

cv2.CAP_PROP_FRAME_WIDTH / HEIGHT / COUNT : width 정보 / height 정보 / 영상 총 프레임수

cv2.CAP_PROP_FPS : 영상 fps 값

• cv2.imshow(’file’, file)

cv2.imshow(video_file, img) # 창의 제목, 화면에 표시할 프레임 이미지 객체

영상 파일 열기

• cap.release()

cap.release() # 자원 반납

• cap.destroyAllWindows()

cv2.destroyAllWindows()

열린 창(영상, 이미지) 닫기

if cv2.waitKey(1) != -1:
	break               

카메라(저장되어 있는 영상 x)는 끝이 없기 때문에 어떤 키가 눌리면 정지되도록

 

---

• OpenCV참고 코드

  import cv2
  
  cap = cv2.VideoCapture("영상 경로" or 0: 카메라장치번호) # 캡처 객체 생성
  
  # 영상 파일 읽기
  if cap.isOpened():                 # 캡쳐 객체 초기화 확인
      while True:
          ret, img = cap.read()      # 다음 프레임 읽기      --- ②
          if ret:                     # 프레임 읽기 정상
              cv2.imshow(video_file, img) # 화면에 표시  --- ③
              cv2.waitKey(25)            # 25ms 지연(40fps로 가정)   --- ④
          else:                       # 다음 프레임 읽을 수 없슴,
              break                   # 재생 완료
  else:
      print("can't open video.")      # 캡쳐 객체 초기화 실패
  
  # 카메라 프레임 읽기
  if cap.isOpened():                      # 캡쳐 객체 연결 확인
      while True:
          ret, img = cap.read()           # 다음 프레임 읽기
          if ret:
              cv2.imshow('camera', img)   # 다음 프레임 이미지 표시
              if cv2.waitKey(1) != -1:    # 1ms 동안 키 입력 대기 ---②
                  break                   # 아무 키라도 입력이 있으면 중지
          else:
              print('no frame')
              break
  else:
      print("can't open camera.")
  
  cap.release()
  cv2.destroyAllWindows()
  

 

참고링크

OpenCV - 3. 이미지 및 동영상 읽기