使用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)

 

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐