Update dataset.py
This commit is contained in:
parent
43d9e6a132
commit
97665de859
|
|
@ -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):
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue