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