jupyter notebookで上手く起動しません

Question

jupyter notebookでpythonを起動し、以下のコードを入れても、In【*】の状態で動きません。
原因が分からず困っています。
ご教授いただければ幸いです。

import datetime
from multiprocessing import Process, Queue
import os
import time

import cv2
import numpy as np

# カメラのIDを指定。
cam_id = 0

# 画像を保存するかと、測定時間を設定。
SAVE_IMG = True
live_time = 300 # seconds(0 or less is endless)
parent_path = []


start_time = 0
parent_path += [datetime.datetime.now().strftime('%y%m%d_%H%M')]
path = os.path.join(*parent_path)
if SAVE_IMG:
    os.makedirs(path, exist_ok=True)
w, h =1280, 960

def get_img(q):
    """
    カメラからデータを受信する関数。
    """
    print('initialize...')
    try:
        # 初期設定を行う。
        cap = cv2.VideoCapture(cam_id,cv2.CAP_MSMF)

        cap.set(3, w)
        cap.set(4, h)
        ret = False
        while not ret:
            time.sleep(0.2)
            ret, frame0 = cap.read()
        print(frame0)
        print(frame0.shape)

        t = 0
        print('start')

        start_time = datetime.datetime.now()
        i = 0

        # カメラからデータを受信しつづけるループ。
        while True:
            end_time = datetime.datetime.now()
            ret, frame = cap.read()
            #ret, frame = True, frame0
            if not ret or frame is None:
                continue
            
            # 測定時間が終わったらループを抜ける。
            if live_time > 0 and end_time.timestamp() - start_time.timestamp() > live_time:
                while not q.empty():
                    time.sleep(1)
                print("end")
                cap.release()
                if SAVE_IMG:
                    with open(os.path.join(path,"settings.txt"), "w") as f:
                        f.write(f"livetime{live_time}\nstart{start_time}\nend {end_time}\n")
                return 0
            if ret:
                i += 1
                print(f"ret{i}")
                q.put((i, frame))
    
    # C-cまたはエラーで終了。
    except KeyboardInterrupt:
        cap.release()
    except Exception as e:
        print(e)
        cap.release()

def show_img(q,qf):
    """
    画像データを保存する。
    """
    t = 0
    g = 1
    f = 0
    show=True
    fname='frames'
    frame_b = np.zeros((1,))
    frame_s = np.zeros((1,))
    try:
        while True:
            if q.empty():
                time.sleep(0.01)
                continue
            i, frame = q.get()
            if frame.size < 2:
                print("reset")
                frame_b = np.zeros_like(frame_b)
                #frame_b += 255
                continue
            print(f"qsize:{qf.qsize()}/{q.qsize()}, max={np.amax(frame)}")

            # ファイル名はフレームごとの連番。
            if SAVE_IMG and np.amax(frame) > 10:
                cv2.imwrite(os.path.join(path, f'{i}.png'), frame)

            if g>0:
                frame = np.average(frame, axis=2)
                frame = frame.astype(np.uint8)

            if show:
                qf.put((i, frame_b))
            t = time.time()
            if np.amax(frame) < 10:
                continue
            #frame[frame<10] = 255

            if frame_b.shape != frame.shape:
                frame_b = frame
            print(np.amax(frame))
            frame_b = np.maximum(frame, frame_b)
    except KeyboardInterrupt:
        cv2.destroyAllWindows()
        return None
    except Exception as e:
        print(e)

def draw_img(q,qf):
    """
    重ね合わせた画像をリアルタイムで表示する。
    実行中にRキーで表示を初期化。
    """
    cv2.namedWindow("frame", cv2.WINDOW_NORMAL)
    t = 0
    show_max = True
    frame_b = np.array([])

    while qf.empty():
        time.sleep(0.01)
    print("start")
    last_i = 0
    start_time = datetime.datetime.now().timestamp()
    try:
        while True:
            if qf.empty():
                time.sleep(0.01)
                continue
            else:
                t = time.time()
                i, frame = qf.get()
            if i < last_i:
                continue
            last_i = i
            if show_max and frame_b.shape == frame.shape:
                frame_b = np.maximum(frame, frame_b)
            else:
                frame_b = frame
            frame_s = frame_b.copy()
            fps= i / max(1,datetime.datetime.now().timestamp() - start_time)
            # 現在のフレームレートを表示する画像に書き込み。
            cv2.putText(frame_s,text=f'{fps:.2f} fps', org=(40,40),fontFace=cv2.FONT_HERSHEY_SIMPLEX,fontScale=1, color=(255,255,255))
            cv2.imshow('frame',frame_s)#cv2.cvtColor(frame, cv2.COLOR_YUV420p2BGR))
            if cv2.waitKey(1) == ord('r'):
                frame_b = np.zeros_like(frame)
                #q.put(np.ndarray([0,]))
    except KeyboardInterrupt:
        cv2.destroyAllWindows()
        return None
    
    except Exception as e:
        print(e)
        cv2.destroyAllWindows()

if __name__ == '__main__':
    # 高速化のため6重に起動。
    try:
        q = Queue(1000)
        qf = Queue(10)
        for n in range(6):
            p = Process(target=show_img, args=(q,qf))
            p.start()
        pd = Process(target=draw_img, args=(q,qf))
        pd.start()
        get_img(q)
        p.terminate()
        pd.terminate()
    except Exception as e:
        print(e)
        p.terminate()
        pd.terminate()

Comments

[meg_]

動作しないのは質問のコードだけですか？

[yk0918]

そうです