liwenyun
/
TGR
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
dataset.py
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):
@ -117,4 +117,4 @@ class CNNDataset(data.Dataset):
img = transforms.Resize((299,299))(img) img = transforms.Resize((299,299))(img)
img = transforms.Compose([transforms.ToTensor()])(img) img = transforms.Compose([transforms.ToTensor()])(img)
#print(img.shape) #print(img.shape)
return img, class_id, class_name, image_name return img, class_id, class_name, image_name