数据处理时,np.ndarray和torch.tensor的纠缠

1. Pytorch

1.1 数据类型

Pytorch的数据预处理torchvision.transforms里面的函数,例如RandomSizedCrop(),全部要求的输入类型为PIL Image,而我们的数据用cv2.imread()读出来的格式时np.ndarray(注意:如果是图片的话,size=(H, W , C), GBR),所以,需要通过Image.fromarray(),将其转化为PIL Image。(如果是BGR的话,要做GBR2RGB的转换)

数据预处理后,直接用torchvision.transforms.ToTensor(),把PIL Image转为[0.0, 1.0]的tensor便可直接输入网络。如果还要对数据做normalization的话,需在ToTensor之后。

如果单纯做array和tensor之间的转换,不做任何其他处理的话,则用torch.from_numpy()和x.numpy()。一般tensor2array时,要执行x.detach().numpy(),因为tensor含梯度信息。

cv2.imshow()和plt.imshow()的区别,cv2的输入应该是大小为(H, W, C)的BGR图片(np.ndarray),而plt的输入是大小为(H, W, C)的RGB图片(np.ndarray),一般用plt.imshow,因为cv2.imshow好慢好慢。。。。

未完待续。。。

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值