We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 51b8482 commit 48ea77cCopy full SHA for 48ea77c
SDT_V3/Classification/Model_Large/util/datasets.py
@@ -22,18 +22,6 @@ def build_dataset(is_train, args):
22
dataset = datasets.ImageFolder(root, transform=transform)
23
return dataset
24
25
-def build_dataset_full(is_train,args):
26
- transform = build_transform(is_train, args)
27
-
28
- train_set = datasets.ImageFolder(root=os.path.join(args.data_path, 'train'),
29
- transform=transform)
30
- valid_set = datasets.ImageFolder(root=os.path.join(args.data_path, 'val'),
31
32
33
- full_set = torch.utils.data.ConcatDataset([train_set, valid_set])
34
- print(full_set)
35
- return full_set
36
37
def build_transform(is_train, args):
38
mean = IMAGENET_DEFAULT_MEAN
39
std = IMAGENET_DEFAULT_STD
0 commit comments