add model arts config

Author: Mark-ZhouWX

official/cv/segment-anything/configs/coco_box_finetune.yaml

@@ -5,7 +5,7 @@ device: Ascend
 mode: 0  # 0: graph, 1: pynative
 work_root: &work_root ./work_dir/
 log_level: info
-amp_level: O2
+amp_level: O0
 
 # ---------------------------------------------
 # Part2: module setting

official/cv/segment-anything/segment_anything/utils/config.py

@@ -32,6 +32,19 @@ def parse_args(parser_config):
     if args_cmd.override_cfg is not None:
         args.merge_with_dotlist(args_cmd.override_cfg)
 
+    # model arts
+    if 'enable_modelarts' in args_cmd and args_cmd.enable_modelarts:
+        print(f'model arts enabled')
+        # output
+        args.work_root = args_cmd.train_url
+        # input
+        args.train_loader.dataset.data_dir = os.path.join(args_cmd.data_url, 'train2017')
+        args.train_loader.dataset.annotation_path = os.path.join(args_cmd.data_url, "annotations", "instances_train2017.json")
+
+        args.eval_loader.dataset.data_dir = os.path.join(args_cmd.data_url, 'val2017')
+        args.eval_loader.dataset.annotation_path = os.path.join(args_cmd.data_url, "annotations",
+                                                                 "instances_val2017.json")
+
     return args
 
 

official/cv/segment-anything/train.py

@@ -1,4 +1,5 @@
 import argparse
+import ast
 
 import mindspore as ms
 from mindspore import amp
@@ -65,5 +66,11 @@ def main(args) -> None:
                                     "For dict, use key=value format, eg: device=False. "
                                     "For nested dict, use '.' to denote hierarchy, eg: optimizer.weight_decay=1e-3."
                                     "For list, use number to denote position, eg: callback.1.interval=100.")
+
+    # model arts
+    parser_config.add_argument("--enable_modelarts", type=ast.literal_eval, default=False)
+    parser_config.add_argument("--train_url", type=str, default="", help="obs path to output folder")
+    parser_config.add_argument("--data_url", type=str, default="", help="obs path to dataset folder")
+
     args = parse_args(parser_config)
     main(args)