Skip to content

Commit 48ea77c

Browse files
committed
Update datasets.py
1 parent 51b8482 commit 48ea77c

File tree

1 file changed

+0
-12
lines changed

1 file changed

+0
-12
lines changed

SDT_V3/Classification/Model_Large/util/datasets.py

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -22,18 +22,6 @@ def build_dataset(is_train, args):
2222
dataset = datasets.ImageFolder(root, transform=transform)
2323
return dataset
2424

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-
transform=transform)
32-
33-
full_set = torch.utils.data.ConcatDataset([train_set, valid_set])
34-
print(full_set)
35-
return full_set
36-
3725
def build_transform(is_train, args):
3826
mean = IMAGENET_DEFAULT_MEAN
3927
std = IMAGENET_DEFAULT_STD

0 commit comments

Comments
 (0)