Moore-AnimateAnyone
Moore-AnimateAnyone copied to clipboard
different transforms in preprocess training data
I noticed that different transformation operations are used for pose and image. For pose: self.cond_transform = transforms.Compose( [ transforms.RandomResizedCrop( self.img_size, scale=self.img_scale, ratio=self.img_ratio, interpolation=transforms.InterpolationMode.BILINEAR, ), transforms.ToTensor(), ] ) For image: self.transform = transforms.Compose( [ transforms.RandomResizedCrop( self.img_size, scale=self.img_scale, ratio=self.img_ratio, interpolation=transforms.InterpolationMode.BILINEAR, ), transforms.ToTensor(), transforms.Normalize([0.5], [0.5]), ] ) Why does pose not require the final normalization step?