yolo 커스텀 학습1 [OpenCV] 동영상에서 이미지 추출하기 YOLO를 이용한 커스텀 학습을 위해서는 이미지 데이터를 만들어야 한다. 우선 파이썬과 opencv가 설치되어 있어야 한다. import cv2 count = 1 test = 1 train = 1 for i in range(1,4): vidcap = cv2.VideoCapture("D:/video/%d.mp4" %i) print("%d번째 영상 시작" %i) while(vidcap.isOpened()): ret, image = vidcap.read() if(ret==False): print("%d번째 영상 끝" %i) print("train=%d" %train) print("test=%d" %test) break if(int(vidcap.get(1)) % 5 == 0): num=count % 10 if .. 2022. 3. 10. 이전 1 다음