PYNQでリアルタイムエッジ検出

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