opencv图片格式转pytorch
使用opencv读图片格式【h*w*c】转为【c*h*w】img = cv2.imread('../imgs/86.jpg',)img=numpy.transpose(img,(2,0,1))input=torch.from_numpy(img)input=torch.tensor(input,dtype=torch.float)...
·
使用opencv读图片
格式【h*w*c】
转为【c*h*w】
img = cv2.imread('../imgs/86.jpg',)
img=numpy.transpose(img,(2,0,1))
input=torch.from_numpy(img)
input=torch.tensor(input,dtype=torch.float)
更多推荐
所有评论(0)