core: dataset: remove unused parameter

girishmk0602 · Jun 19, 2020 · 5a8d591 · 5a8d591
1 parent 2de5f4e
commit 5a8d591
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/dataset.py b/core/dataset.py
@@ -51,12 +51,12 @@ def __init__(self, dataset_type):
         self.anchor_per_scale = cfg.YOLO.ANCHOR_PER_SCALE
         self.max_bbox_per_scale = 150
 
-        self.annotations = self.load_annotations(dataset_type)
+        self.annotations = self.load_annotations()
         self.num_samples = len(self.annotations)
         self.num_batchs = int(np.ceil(self.num_samples / self.batch_size))
         self.batch_count = 0
 
-    def load_annotations(self, dataset_type):
+    def load_annotations(self):
         with open(self.annot_path, "r") as f:
             txt = f.readlines()
             annotations = [