minor fixes

ykotseruba · Sep 20, 2019 · 0e819c6 · 0e819c6
1 parent 4c23d1e
commit 0e819c6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/jaad_data.py b/jaad_data.py
@@ -827,11 +827,11 @@ def get_detection_data(self, image_set, method, occlusion_type=None, file_path='
         if method == 'frcnn':
             return self._get_data_frcnn(ped_samples)
         elif method == 'retinanet':
-            self._generate_csv_data_retinanet(image_set, file_path, ped_samples)
+            return self._generate_csv_data_retinanet(image_set, file_path, ped_samples)
         elif method == 'yolo3':
-            self._generate_csv_data_yolo3(image_set, file_path, ped_samples)
+            return self._generate_csv_data_yolo3(image_set, file_path, ped_samples)
         elif method == 'ssd':
-            self._generate_csv_data_ssd(image_set, file_path, ped_samples)
+            return self._generate_csv_data_ssd(image_set, file_path, ped_samples)
 
     def _get_data_frcnn(self, ped_samples):
         """