?
原因:由于paddle暂时不支持在子进程中进行GPU Tensor的操作,所以在预处理时候,不能使用totenso()这个算子,原论文中的pytorch的预处理算子
?由于paddle无对应的ToPILImage,所以我参考了对应实现PyTorch-PaddlePaddle API映射表-API文档-PaddlePaddle深度学习平台
转换后paddle对应的写法
?这里要特别注意,mean=[0., 0., 0.], std=[255.0, 255.0, 255.0]必须写成浮点数,不能用整数,否则产生的图像会与原图像有偏差。
def get_test_transforms(configs):
transforms_test = transforms.Compose([
ToPILImage(),
transforms.Resize((configs.DATA.imgH, configs.DATA.imgW)), # HWC
transforms.Transpose(order=(2, 0, 1)), # CHW
transforms.Normalize(mean=[0., 0., 0.], std=[255.0, 255.0, 255.0]),
transforms.Normalize(mean=[0.5, 0.5, 0.5], std=[0.5, 0.5, 0.5])])
return transforms_test
原图:
锐化后的图像(std,mean是整数):
??
?
|