Python_Pytorch比较好用的函数

1、.strip( )移除首尾空字符

strip('0'),移除字符串首尾的字符0

2、os.path.basename() 返回path最后的文件名，若path以/或\结尾，那么就会返回空值。

3、torchvision.utils.save_image直接保存tensor为图片

这种方式只能保存RGB彩色图像，如果网络的输出是单通道灰度图像，则该函数依然会输出三个通道，每个通道的数值都是相同的，即“伪灰度图像”，虽然从视觉效果上看不出区别，但是图像所占内存比正常情况大了两倍。