파이썬 실시간 병렬처리

Question

yolo 프레임 향상을 위해 실시간으로 들어오는 프레임을 병렬처리하려고 합니다 프레임이 들어올때마다 프로세스를 생성해서 처리 후 출력하려 하는데 어떻게 구현해야할까요..?multiprocessing을 사용해 아래와 같이 구현하려했지만 이전 같은 함수는 이전 프로세스가 끝나야 실행되는 것 같아 실패했습니다..import time from multiprocessing import Process, Manager import cv2 from ultralytics import YOLO def process_frame(model, frame, processed_frames): inputs &#61; [frame] result &#61; model(inputs)[0] plot &#61; result.plot() processed_frames.put(plot) def show_frame(processed_frames): while True: if processed_frames.qsize() > 0: frame &#61; processed_frames.get() cv2.imshow('result', frame) k &#61; cv2.waitKey(1) & 0xFF if k &#61;&#61; 27: break if __name__ &#61;&#61; &#34;__main__&#34;: model &#61; YOLO('yolov8n.pt') cap &#61; cv2.VideoCapture(1) # set video device #device number if cap.isOpened(): # check device connection print('width: {}, height : {}'.format(cap.get(3), cap.get(4))) # print input resolution m &#61; Manager() processed_frames &#61; m.Queue() Process(target&#61;show_frame, args&#61;(processed_frames,)).start() while True: # loop ret, frame &#61; cap.read() # get frame if ret: # check if frame has been grabbed frame &#61; cv2.flip(frame, 1) # flip horizontal print('start: %f' % time.time()) p &#61; Process(target&#61;process_frame, args&#61;(model, frame, processed_frames)) p.start() print('started: %f' % time.time()) print('a') else: print('error') break cap.release() cv2.destroyAllWindows()

hmmm · Answer

제가 예시로 만들어본 코드 입니다. 이 방식대로 하면 문제 해결하는데 도움이 될 것 같아 올려봅니다~

hmmm · Answer

import time from multiprocessing import Queue, Process import cv2 class Multi() : def __init__(self): self.q &#61; Queue() def webcam(self): cap &#61; cv2.VideoCapture(0) if cap.isOpened(): while True: ret, frame &#61; cap.read() if ret: self.q.put(frame) cv2.imshow('frame', frame) key &#61; cv2.waitKey(10) # 종료 키 if key &#61;&#61; ord('q') or key &#61;&#61; ord('Q'): break else: print('error') break cap.release() cv2.destroyAllWindows() def printShape(self): while 1 : data &#61; self.q.get() print(data.shape) if __name__ &#61;&#61; &#34;__main__&#34;: M &#61; Multi() p1 &#61; Process(target&#61;M.webcam) p2 &#61; Process(target&#61;M.printShape) p1.start() p2.start()

파이썬 실시간 병렬처리

2개의 답변

#yolo 목록