hyp embeddings labels only for b200

Author: ankitdhall

network/order_embeddings_h.py

@@ -13,6 +13,8 @@
 from network.evaluation import MultiLabelEvaluation, Evaluation, MultiLabelEvaluationSingleThresh, MultiLevelEvaluation
 from network.finetuner import CIFAR10
 
+from data.db import Butterfly200LabelMap
+
 from data.db import ETHECLabelMap, ETHECDB, ETHECDBMerged, ETHECLabelMapMerged, ETHECLabelMapMergedSmall, ETHECDBMergedSmall
 from network.loss import MultiLevelCELoss, MultiLabelSMLoss, LastLevelCELoss, MaskedCELoss, HierarchicalSoftmaxLoss
 
@@ -1352,11 +1354,11 @@ def order_embedding_train_model(arguments):
 
     # initial_crop = 324
     input_size = 224
-    labelmap = ETHECLabelMap()
+    labelmap = Butterfly200LabelMap()
     if arguments.merged:
-        labelmap = ETHECLabelMapMerged()
+        labelmap = Butterfly200LabelMap()
     if arguments.debug:
-        labelmap = ETHECLabelMapMergedSmall()
+        labelmap = Butterfly200LabelMap()
 
     train_data_transforms = transforms.Compose([transforms.ToPILImage(),
                                                 transforms.Resize((input_size, input_size)),
@@ -1381,34 +1383,34 @@ def order_embedding_train_model(arguments):
                                                        ])
 
     if not arguments.merged:
-        train_set = ETHECDB(path_to_json='../database/ETHEC/train.json',
+        train_set = ETHECDB(path_to_json='../database/butterfly200/train.json',
                             path_to_images=arguments.image_dir,
                             labelmap=labelmap, transform=train_data_transforms, with_images=False)
-        val_set = ETHECDB(path_to_json='../database/ETHEC/val.json',
+        val_set = ETHECDB(path_to_json='../database/butterfly200/val.json',
                           path_to_images=arguments.image_dir,
                           labelmap=labelmap, transform=val_test_data_transforms, with_images=False)
-        test_set = ETHECDB(path_to_json='../database/ETHEC/test.json',
+        test_set = ETHECDB(path_to_json='../database/butterfly200/test.json',
                            path_to_images=arguments.image_dir,
                            labelmap=labelmap, transform=val_test_data_transforms, with_images=False)
     elif not arguments.debug:
-        train_set = ETHECDBMerged(path_to_json='../database/ETHEC/train.json',
+        train_set = ETHECDBMerged(path_to_json='../database/butterfly200/train.json',
                                   path_to_images=arguments.image_dir,
                                   labelmap=labelmap, transform=train_data_transforms, with_images=False)
-        val_set = ETHECDBMerged(path_to_json='../database/ETHEC/val.json',
+        val_set = ETHECDBMerged(path_to_json='../database/butterfly200/val.json',
                                 path_to_images=arguments.image_dir,
                                 labelmap=labelmap, transform=val_test_data_transforms, with_images=False)
-        test_set = ETHECDBMerged(path_to_json='../database/ETHEC/test.json',
+        test_set = ETHECDBMerged(path_to_json='../database/butterfly200/test.json',
                                  path_to_images=arguments.image_dir,
                                  labelmap=labelmap, transform=val_test_data_transforms, with_images=False)
     else:
         labelmap = ETHECLabelMapMergedSmall(single_level=False)
-        train_set = ETHECDBMergedSmall(path_to_json='../database/ETHEC/train.json',
+        train_set = ETHECDBMergedSmall(path_to_json='../database/butterfly200/train.json',
                                        path_to_images=arguments.image_dir,
                                        labelmap=labelmap, transform=train_data_transforms, with_images=False)
-        val_set = ETHECDBMergedSmall(path_to_json='../database/ETHEC/val.json',
+        val_set = ETHECDBMergedSmall(path_to_json='../database/butterfly200/val.json',
                                      path_to_images=arguments.image_dir,
                                      labelmap=labelmap, transform=val_test_data_transforms, with_images=False)
-        test_set = ETHECDBMergedSmall(path_to_json='../database/ETHEC/test.json',
+        test_set = ETHECDBMergedSmall(path_to_json='../database/butterfly200/test.json',
                                       path_to_images=arguments.image_dir,
                                       labelmap=labelmap, transform=val_test_data_transforms, with_images=False)