测试opencv/pytorch/tensorflow是否安装成功

158 阅读1分钟

opencv

# 调用摄像头
# 打开摄像头并灰度化显示
import cv2

capture = cv2.VideoCapture(0)
while(True):
    # 获取一帧
    ret, frame = capture.read()
    # 将这帧转换为灰度图
#    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    cv2.imshow('frame', frame)
    if cv2.waitKey(1) == ord('q'):
        break

pytorch

import torch

print(torch.__version__)

tensorflow

import tensorflow as tf
from tensorflow.keras import layers
print(tf.__version__)
print(tf.keras.__version__)

附:知乎TensorFlow2.0教程