canny.py
from pynq import Overlay Overlay("base.bit").download() import time import cv2 import numpy as np from pynq.drivers import HDMI from pynq.drivers.video import VMODE_1920x1080 hdmi_out = HDMI('out', video_mode=VMODE_1920x1080) hdmi_in = HDMI('in', init_timeout=10, frame_list=hdmi_out.frame_list) hdmi_in.start() hdmi_out.start() for i in range(10): print("waiting ",10-i) time.sleep(1) print("change HDMI output frame") hdmi_out.frame_index_next() print("main loop") try: while True: frame = hdmi_in.frame_raw() np_frame= (np.frombuffer(frame, dtype=np.uint8)).reshape(1080,1920,3) edges = cv2.Canny(np_frame,100,200) stacked_img = np.stack((edges,)*3,axis=-1) hdmi_out.frame_raw(bytearray(stacked_img.tobytes())) except KeyboardInterrupt: hdmi_out.stop() hdmi_in.stop() del hdmi_in, hdmi_out
実行
$ sudo python3.6 canny.py