Spaces:
Running
Running
| import torchvision.transforms as transforms | |
| im_mean = (124, 116, 104) | |
| im_normalization = transforms.Normalize( | |
| mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225] | |
| ) | |
| inv_im_trans = transforms.Normalize( | |
| mean=[-0.485 / 0.229, -0.456 / 0.224, -0.406 / 0.225], | |
| std=[1 / 0.229, 1 / 0.224, 1 / 0.225], | |
| ) | |