1、.strip( )移除首尾空字符
strip('0'),移除字符串首尾的字符0
2、os.path.basename() 返回path最后的文件名,若path以/或\结尾,那么就会返回空值。
3、torchvision.utils.save_image直接保存tensor为图片
这种方式只能保存RGB彩色图像,如果网络的输出是单通道灰度图像,则该函数依然会输出三个通道,每个通道的数值都是相同的,即“伪灰度图像”,虽然从视觉效果上看不出区别,但是图像所占内存比正常情况大了两倍。
1、.strip( )移除首尾空字符
strip('0'),移除字符串首尾的字符0
2、os.path.basename() 返回path最后的文件名,若path以/或\结尾,那么就会返回空值。
3、torchvision.utils.save_image直接保存tensor为图片
这种方式只能保存RGB彩色图像,如果网络的输出是单通道灰度图像,则该函数依然会输出三个通道,每个通道的数值都是相同的,即“伪灰度图像”,虽然从视觉效果上看不出区别,但是图像所占内存比正常情况大了两倍。