Moore-AnimateAnyone icon indicating copy to clipboard operation
Moore-AnimateAnyone copied to clipboard

different transforms in preprocess training data

Open chlinfeng1997 opened this issue 2 years ago • 0 comments

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?

chlinfeng1997 avatar Jan 18 '24 10:01 chlinfeng1997