update domainnet.py

Author: weigq

datasets/domainnet.py

@@ -72,55 +72,5 @@ def __init__(self, data_root, domains: list, status: str = 'train', trim: int =
 
         assert len(self.domain_id) == len(self.data), f'domain_id not match data'
 
-        # self.train = True if status == 'train' else False
-        # self.num_domain = len(domains)
-        #
-        # # read txt files
-        # data = []
-        # len_domains = []
-        # for _domain in domains:
-        #     suffix = 'train' if status == 'train' else 'test'
-        #     _data = read_images_labels(
-        #         os.path.join(f'dataset_map/domainnet', f'{_domain}_{suffix}.txt'),
-        #         shuffle=(status == 'train'),
-        #         trim=trim
-        #     )
-        #     len_domains.append(len(_data))
-        #     data.append(_data)
-        #
-        # max_len = max(len_domains)
-        # # keep all domains have same # data | training
-        # if status == 'train':
-        #     for i in range(len(data)):
-        #         data[i] = data[i] * round(.5 + max_len / len(data[i]))
-        #     self.data = data
-        # else:
-        #     # cat to one domain
-        #     self.data = [functools.reduce(operator.iconcat, data, [])]
-        #
-        #     domain_id = [[i] * len_domains[i] for i in range(self.num_domain)]
-        #     self.domain_id = [functools.reduce(operator.iconcat, domain_id, [])]
-        #
-        #     assert len(self.domain_id) == len(self.data), f'domain_id not match data'
-    #
-    # def __getitem__(self, index):
-    #     if self.train:
-    #         domain = random.randint(0, self.num_domain - 1)
-    #         path, label = self.data[domain][index]
-    #     else:
-    #         domain = self.domain_id[0][index]
-    #         path, label = self.data[0][index]
-    #     path = os.path.join(self.image_root, path)
-    #     with Image.open(path) as image:
-    #         image = image.convert('RGB')
-    #     if self.transform is not None:
-    #         image = self.transform(image)
-    #
-    #     return {
-    #         'image': image,
-    #         'label': label,
-    #         'domain': domain
-    #     }
-
     def __len__(self):
         return len(self.data)