Changed name of camera trap dataset class to be consistent with others

microsoft · Mar 18, 2019 · 23c33a4 · 23c33a4
1 parent 9b8bc06
commit 23c33a4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/FasterRCNNDetection/data/coco_camera_traps_dataset.py b/FasterRCNNDetection/data/coco_camera_traps_dataset.py
@@ -42,7 +42,7 @@ def load_taxonomy(ann_data, tax_levels, classes):
     return taxonomy, classes_taxonomic
 
 
-class CocoCameraTrapsBBoxDataset:
+class CocoCameraTrapsBboxDataset:
     """Bounding box dataset
 
     The index corresponds to each image.

diff --git a/FasterRCNNDetection/data/dataset.py b/FasterRCNNDetection/data/dataset.py
@@ -9,7 +9,7 @@
 ################
 
 import torch as t
-from .coco_camera_traps_dataset import CocoCameraTrapsBBoxDataset
+from .coco_camera_traps_dataset import CocoCameraTrapsBboxDataset
 from .voc_dataset import VOCBboxDataset
 from .vott_dataset import VottBboxDataset
 from .iwildcam_dataset import IWildCamBboxDataset
@@ -170,7 +170,7 @@ def __init__(self, opt, class_names = [], use_difficult=True):
         if opt.dataset == 'voc':
           self.db = VOCBboxDataset(opt.voc_data_dir, split='test', use_difficult=use_difficult)
         elif opt.dataset in ['multiclass', 'oneclass']:
-          self.db = CocoCameraTrapsBboxDatasetBboxDataset(opt.image_root, opt.val_annotation)
+          self.db = CocoCameraTrapsBboxDataset(opt.image_root, opt.val_annotation)
         elif opt.dataset == 'vott':
             self.db = VottBboxDataset(opt.val_image_dir, class_names)
         elif opt.dataset == 'seals':