파이썬 실시간 영상 스트리밍 관련 질문입니다!

코드는 이렇게 되어있습니다!

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
class Camera:
    def __init__(self, source, PT_PATH):
        self.capture = cv2.VideoCapture(source)
        if not PT_PATH == "original":
            self.model = load_model(PT_PATH)
        else:
            self.model = "original"
        self.streaming_state = False
        self.frame_queue = queue.Queue()
        self.thread = threading.Thread(target=self.read_frames, args=())

    def start(self):
        self.streaming_state = True
        self.thread.start()

    def stop(self):
        self.streaming_state = False
        while not self.frame_queue.empty():
            try:
                self.frame_queue.get_nowait()
            except queue.Empty:
                break
        self.capture.release()
        print("메모리에서 해제된 후 상태 : ",self.capture.isOpened())


    def read_frames(self):
        while self.streaming_state:
            # print("capture 상태 : ",self.capture.isOpened())
            success, capture_img = self.capture.read()
            if not success:
                print("더이상 읽어올 Frame이 존재하지 않습니다.")
                break
            
            if not self.model == "original":
                pre_process_image, model_to_device = preprocess_frame(capture_img=capture_img, model=self.model)
                post_process_image = postprocess_frame(pre_process_image, capture_img, model_to_device)
                self.frame_queue.put((b'--frame\r\n'
                        b'Content-Type: image/jpeg\r\n\r\n' + post_process_image + b'\r\n'))
            else:
                capture_img = httpProcessing(capture_img=capture_img)
                self.frame_queue.put((b'--frame\r\n'
                        b'Content-Type: image/jpeg\r\n\r\n' + capture_img + b'\r\n'))

    
    def get_frame(self):
        while True:
            if self.frame_queue.empty():
                continue
            yield self.frame_queue.get()