历时一小时的愁云惨淡万里凝。 这个问题终于被我找到啦啦啦啦
我的图像是经过以下处理得到tensor格式,根据上面我提出的无法检测到人脸的问题:
img_pil = Image.open(random.choice(self.out_domain_hairstyle_img_path_list))
self.image_transform = transforms.Compose([transforms.ToTensor(),transforms.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])])
解决办法:
img_transforms = transforms.Compose([transforms.Resize((256, 256)), transforms.Normalize([-1, -1, -1], [1 / 0.5, 1 / 0.5, 1 / 0.5])])
input_image = img_transforms(x.squeeze(dim=0).float()) * 255.0
# 转换为 uint8
input_image = input_image.permute(1, 2, 0).byte().cpu().numpy()
有用的话,求个赞