assign command line argument to variable

Author: yharby

deep-learning-datasets-maker/split_rs_data.py

@@ -37,6 +37,7 @@
 import os
 import os.path as osp
 from .utils import *
+import argparse
 
 
 class SplitRSData:
@@ -333,9 +334,49 @@ def run(self):
             else :
                 feedback.pushInfo("Option instance segmentation is not selected")
 
-            # if self.dlg.checkBoxPaddle.isChecked():
-                # generate_list()
-            # else :
-                # feedback.pushInfo("Option instance segmentation is not selected")
-
-            iface.messageBar().pushMessage("You will find the dataset in " + image_folder_path, level=Qgis.Success, duration=5)
+            if self.dlg.checkBoxPaddle.isChecked():
+                    dataset_path = "/".join(image_folder_path.split("/")[:-1])
+                    Training_Set = self.dlg.mOpacityWidget_Training.opacity()
+                    Val_Set =  self.dlg.mOpacityWidget_Validating.opacity()
+                    Testing_Set = self.dlg.mOpacityWidget_Testing.opacity()
+                    def parse_args(dataset_path ,image_folder_path, label_folder_path, Training_Set, Val_Set, Testing_Set):
+                        parser = argparse.ArgumentParser(
+                            description=
+                            'A tool for proportionally randomizing dataset to produce file lists.')
+                        parser.add_argument('dataset_root', help='the dataset root path', type=str, default=dataset_path)
+                        parser.add_argument(
+                            'images_dir_name', help='the directory name of images', type=str, default=image_folder_path)
+                        parser.add_argument(
+                            'labels_dir_name', help='the directory name of labels', type=str, default=label_folder_path)
+                        parser.add_argument(
+                            '--split', help='', nargs=3, type=float, default=[Training_Set, Val_Set, Testing_Set])
+                        parser.add_argument(
+                            '--label_class',
+                            help='label class names',
+                            type=str,
+                            nargs='*',
+                            default=['__background__', '__foreground__'])
+                        parser.add_argument(
+                            '--separator',
+                            dest='separator',
+                            help='file list separator',
+                            default=" ",
+                            type=str)
+                        parser.add_argument(
+                            '--format',
+                            help='data format of images and labels, e.g. jpg, tif or png.',
+                            type=str,
+                            nargs=2,
+                            default=['jpg', 'png'])
+                        parser.add_argument(
+                            '--postfix',
+                            help='postfix of images or labels',
+                            type=str,
+                            nargs=2,
+                            default=['', ''])
+
+                        return parser.parse_args()
+                    args = parse_args(dataset_path ,image_folder_path, label_folder_path, Training_Set, Val_Set, Testing_Set)
+                    generate_list(args)
+
+            iface.messageBar().pushMessage("You will find the dataset in " + dataset_path, level=Qgis.Success, duration=5)

deep-learning-datasets-maker/utils/__init__.py

@@ -1,4 +1,4 @@
 from .rasterize import rasterize
 from .splitting import splitting
 from .semantic2instance import segMaskB2I
-from .paddlepaddle_split_dataset_list import pp_generate_list
+from .paddlepaddle_split_dataset_list import generate_list

deep-learning-datasets-maker/utils/paddlepaddle_split_dataset_list.py

@@ -15,49 +15,11 @@
 
 import glob
 import os.path
-import argparse
+# import argparse
 import warnings
 import numpy as np
 
 # TODO:  assign command line argument to variable
-def parse_args():
-    parser = argparse.ArgumentParser(
-        description=
-        'A tool for proportionally randomizing dataset to produce file lists.')
-    parser.add_argument('dataset_root', help='the dataset root path', type=str)
-    parser.add_argument(
-        'images_dir_name', help='the directory name of images', type=str)
-    parser.add_argument(
-        'labels_dir_name', help='the directory name of labels', type=str)
-    parser.add_argument(
-        '--split', help='', nargs=3, type=float, default=[0.7, 0.3, 0])
-    parser.add_argument(
-        '--label_class',
-        help='label class names',
-        type=str,
-        nargs='*',
-        default=['__background__', '__foreground__'])
-    parser.add_argument(
-        '--separator',
-        dest='separator',
-        help='file list separator',
-        default=" ",
-        type=str)
-    parser.add_argument(
-        '--format',
-        help='data format of images and labels, e.g. jpg, tif or png.',
-        type=str,
-        nargs=2,
-        default=['jpg', 'png'])
-    parser.add_argument(
-        '--postfix',
-        help='postfix of images or labels',
-        type=str,
-        nargs=2,
-        default=['', ''])
-
-    return parser.parse_args()
-
 
 def get_files(path, format, postfix):
     pattern = '*%s.%s' % (postfix, format)
@@ -146,6 +108,6 @@ def generate_list(args):
             start = end
 
 
-if __name__ == '__main__':
-    args = parse_args()
-    generate_list(args)
+# if __name__ == '__main__':
+#     args = parse_args()
+#     generate_list(args)