分类：读入图像

2020
10-01

Pytorch 使用opnecv读入图像由HWC转为BCHW格式方式

传统使用opencv自带的swapaxes进行转换，然后使用pytorch的from_numpy转为tensor例如：img=img.swapaxes(1,2).swapaxes(0,1)但是有一个常用的图像增广模块albumentations模块中的img_to_tensor进行直接转换注意：我这里使用unsqueeze是为了验证模型的输出，如果使用pytorch的dataset则不需要使用这个操作补充知识：pytorch只用中要注意通道问题cv读进来的是BGR图像，通道是hcw，在torch中使用要注意维度转换def__getitem__(self... 继续阅读 >