fix split_odvg.py

tools/misc/split_odvg.py

@@ -17,8 +17,10 @@ def parse_args():
         'out_dir',
         type=str,
         help='The output directory of coco semi-supervised annotations.')
+    parser.add_argument('--label-map-file', '-m', type=str, help='label map file')
     parser.add_argument(
         '--num-img',
+        '-n',
         default=200,
         type=int,
         help='num of extract image, -1 means all images')
@@ -58,10 +60,18 @@ def main():
         progress_bar.update()
 
     out_path = os.path.join(args.out_dir, args.ann_file)
+    out_dir = os.path.dirname(out_path)
+    mkdir_or_exist(out_dir)
 
     with jsonlines.open(out_path, mode='w') as writer:
         writer.write_all(data_list[:num_img])
 
+    if args.label_map_file is not None:
+        out_dir = os.path.dirname(os.path.join(args.out_dir, args.label_map_file))
+        mkdir_or_exist(out_dir)
+        shutil.copyfile(os.path.join(args.data_root, args.label_map_file),
+                        os.path.join(args.out_dir, args.label_map_file))
+
 
 if __name__ == '__main__':
     main()