问题描述
导入 cv2
hub_handle = 'https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2' hub_module = hub.load(hub_handle)
cap = cv2.VideoCapture(r"/video/a.mp4") frame_size=256
fourc = cv2.VideoWriter_fourcc(*'MP4V') out = cv2.VideoWriter(r"/video/video_a_stylized.mp4",fourc,60.0,(frame_size,frame_size))
计数器 = 0 为真: 打印(f“{计数器}”)
flag,frame=cap.read()
if flag:
image_rgb_np = cv2.cvtColor(frame,cv2.COLOR_BGR2RGB)
print(image_rgb_np)
resized_image_np = resize_image_to_square(image_rgb_np,image_size=(frame_size,frame_size))
outputs = hub_module(tf.constant(resized_image_np),tf.constant(style_image))
stylized_image = outputs[0]
image_pil = tf.keras.preprocessing.image.array_to_img(stylized_image[0])
image_bgr_np = cv2.cvtColor(np.array(image_pil),cv2.COLOR_RGB2BGR)
out.write(image_bgr_np)
else:
break
counter = counter+1
out.release()
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)