utils/constants.py

import numpy as np

class Constants:
    pytorch_mean = np.array([0.485, 0.456, 0.406])
    pytorch_std = np.array([0.229, 0.224, 0.225])