From 2c93ed9375362011a17c1684da89d9dd4a0bb9e4 Mon Sep 17 00:00:00 2001
From: "Wu, Jiantao (PG/R - Comp Sci & Elec Eng)" <jiantao.wu@surrey.ac.uk>
Date: Fri, 3 May 2024 14:47:39 +0000
Subject: [PATCH] add INatDataset
---
vitookit/datasets/build_dataset.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/vitookit/datasets/build_dataset.py b/vitookit/datasets/build_dataset.py
index 34ae035..b64203f 100644
--- a/vitookit/datasets/build_dataset.py
+++ b/vitookit/datasets/build_dataset.py
@@ -78,6 +78,10 @@ def build_dataset(args, is_train, trnsfrm=None,):
trnsfrm.transforms.insert(-2,transforms.Grayscale(num_output_channels=3))
dataset = datasets.Omniglot(args.data_location,transform=tfm,download=True)
nb_classes = 1623
+
+ elif args.data_set == 'INAT':
+ dataset = INatDataset(args.data_path, train=is_train, year=2018,
+ transform=tfm)
else:
print('dataloader of {} is not implemented .. please add the dataloader under datasets folder.'.format(args.data_set))
--
GitLab