Update dataset.py

This commit is contained in:
Jianping Zhang 2023-07-18 16:47:13 +08:00 committed by GitHub
parent 43d9e6a132
commit 97665de859
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions

View File

@ -18,10 +18,10 @@ DEFAULT_CROP_PCT = 0.875 # 0.9, 1.0
IMAGENET_DEFAULT_MEAN = (0.485, 0.456, 0.406) IMAGENET_DEFAULT_MEAN = (0.485, 0.456, 0.406)
IMAGENET_DEFAULT_STD = (0.229, 0.224, 0.225) IMAGENET_DEFAULT_STD = (0.229, 0.224, 0.225)
# Imagenet 21k # Imagenet 21k
#IMAGENET_INCEPTION_MEAN = (0.5, 0.5, 0.5) IMAGENET_INCEPTION_MEAN = (0.5, 0.5, 0.5)
#IMAGENET_INCEPTION_STD = (0.5, 0.5, 0.5) IMAGENET_INCEPTION_STD = (0.5, 0.5, 0.5)
IMAGENET_INCEPTION_MEAN = (0.0, 0.0, 0.0) #IMAGENET_INCEPTION_MEAN = (0.0, 0.0, 0.0)
IMAGENET_INCEPTION_STD = (1.0, 1.0, 1.0) #IMAGENET_INCEPTION_STD = (1.0, 1.0, 1.0)
def params(model_name): def params(model_name):