3-27Pytorch简单编程技巧(模型的保存/加载)

在这里插入图片描述
并行化操作加快数据处理速度
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
安装cv2命令

pip install opencv-python
import torch
import numpy as np
import cv2
data = cv2.imread('test.png')
#print(data)
cv2.imshow('test1',data)
cv2.waitKey(0)
out = torch.from_numpy(data)
out = out.to(torch.device("cuda"))
print(out.is_cuda)
# print(out)
True
out = torch.flip(out,dims=[0])
data = out.numpy()
print(data)
print(out)
print(data)
cv2.imshow('test2',data)
cv2.waitKey(0)

在这里插入图片描述

out = out.to(torch.device("cpu"))
print(out.is_cuda)
False
已标记关键词 清除标记
©️2020 CSDN 皮肤主题: 深蓝海洋 设计师:CSDN官方博客 返回首页