[Python] OpenCV 기초 4 - 이미지 및 동영상 저장

1. 이미지 저장 

- Opencv 라이브러리를 이용하면 이미지를 jpg와 png 형식으로 저장할 수 있다. 

cv2.imwrite([저장할 이름.파일형식], 이미지)

import cv2
img = cv2.imread('img.jpg',cv2.IMREAD_GRAYSCALE) # 흑백으로 불러오기
cv2.imwrite('img_save.png',img) # png 형태로 저장

 

 

2. 동영상 저장

import cv2
cap = cv2.VideoCapture('video.mp4')

#코덱 정의
fourcc = cv2.VideoWriter_fourcc(*'DIVX') # *은 한칸씩 떨어뜨려서 반환

width = round(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
height = round(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
fps = cap.get(cv2.CAP_PROP_FPS) *2 # 영상 속도가 두배

out = cv2.VideoWriter('output_2.avi',fourcc, fps, (width,height))
#저장 파일명, 코덱, FPS, 크기(width, height)


while cap.isOpened():
    ret, frame = cap.read()
    if not ret:
        break
    
    out.write(frame) # 영상 데이터만 저장 (소리 X)
    
    cv2.imshow('video',frame)
    if cv2.waitKey(1) == ord('q'):
        break

out.release()    
cap.release()
cv2.destroyAllWindows()

코덱은 DIVX 형식을 사용했다. *'DIVX' 로 코드를 입력하면 D I V X 로 반환된다. 

cv2.CAP_PROP_FRAME_WIDTH 로 가로길이 추출,  

cv2.CAP_PROP_FRAME_HEIGHT로 세로길이를 추출하여 cv2.VideoWriter()로 비디오를 저장한다. 

 

while문을 이용해 동영상이 재생되는 동안 모든 프레임을 out.write(frame)으로 저장한다.