default class mapping

eg4000 · May 19, 2019 · 5c94f84 · 5c94f84
1 parent abe5d00
commit 5c94f84
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/object_detector_retinanet/keras_retinanet/bin/predict.py b/object_detector_retinanet/keras_retinanet/bin/predict.py
@@ -77,7 +77,7 @@ def parse_args(args):
     csv_parser.add_argument('--annotations', help='Path to CSV file containing annotations for evaluation.',
                             default=args_annotations)
     csv_parser.add_argument('--classes', help='Path to a CSV file containing class label mapping.',
-                            default=os.path.join(os.getcwd(), 'class_mappings.csv'))
+                            default=os.path.join(os.path.dirname(os.path.abspath(__file__)), 'class_mappings.csv'))
     parser.add_argument('--hard_score_rate', help='', default=1.)
 
     parser.add_argument('model', help='Path to RetinaNet model.')

diff --git a/object_detector_retinanet/keras_retinanet/bin/train.py b/object_detector_retinanet/keras_retinanet/bin/train.py
@@ -304,7 +304,7 @@ def csv_list(string):
     csv_parser.add_argument('--annotations', help='Path to CSV file containing annotations for training.',
                             default=args_annotations)
     csv_parser.add_argument('--classes', help='Path to a CSV file containing class label mapping.',
-                            default=os.path.join(os.getcwd(), 'class_mappings.csv'))
+                            default=os.path.join(os.path.dirname(os.path.abspath(__file__)), 'class_mappings.csv'))
     csv_parser.add_argument('--val-annotations',
                             help='Path to CSV file containing annotations for validation (optional).',
                             default=args_val_annotations)

diff --git a/object_detector_retinanet/keras_retinanet/bin/train_iou.py b/object_detector_retinanet/keras_retinanet/bin/train_iou.py
@@ -308,7 +308,7 @@ def csv_list(string):
     csv_parser.add_argument('--annotations', help='Path to CSV file containing annotations for training.',
                             default=args_annotations)
     csv_parser.add_argument('--classes', help='Path to a CSV file containing class label mapping.',
-                            default=os.path.join(os.getcwd(), 'class_mappings.csv'))
+                            default=os.path.join(os.path.dirname(os.path.abspath(__file__)), 'class_mappings.csv'))
     csv_parser.add_argument('--val-annotations',
                             help='Path to CSV file containing annotations for validation (optional).',
                             default=args_val_annotations)