Function final_transform

package_utils/transform.py:103–110

(_cfg)

def final_transform(_cfg):
    return transforms.Compose([
                transforms.ToTensor(),
                transforms.Normalize(
                    mean=_cfg.TRANSFORM.normalize.mean, 
                    std=_cfg.TRANSFORM.normalize.std,
                ),
            ])


def randaffine(img, mask):