Merge pull request #92 from Archaic-Atom/dev

Dev
Archaic-Atom · May 2, 2023 · 371cd51 · 371cd51
2 parents c22eaca + 2c6b088
commit 371cd51
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 8 deletions.
diff --git a/Source/JackFramework/Core/Graph/_meta_ops.py b/Source/JackFramework/Core/Graph/_meta_ops.py
@@ -42,7 +42,8 @@ def _init_ddp_model(self) -> None:
         assert self._model is not None
         for i, model_item in enumerate(self._model):
             model_item = model_item.to(self.rank)
-            self._model[i] = DDP(model_item, device_ids=[self.rank], find_unused_parameters=True)
+            self._model[i] = DDP(model_item, device_ids=[self.rank],
+                                 find_unused_parameters=self.__args.debug)
 
     def _init_dp_model(self) -> None:
         assert self._model is not None

diff --git a/Source/JackFramework/FileHandler/file_handler.py b/Source/JackFramework/FileHandler/file_handler.py
@@ -35,7 +35,7 @@ def close_file(fd_file: object) -> None:
 
     @staticmethod
     def write_file(fd_file: object, data_str: str) -> None:
-        fd_file.write(f"{data_str} \n")
+        fd_file.write(f"{data_str}\n")
         fd_file.flush()
 
     @staticmethod

diff --git a/Source/JackFramework/ImgHandler/data_augmentation.py b/Source/JackFramework/ImgHandler/data_augmentation.py
@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 import random
 import numpy as np
+import cv2
 
 
 class DataAugmentation(object):
@@ -87,3 +88,10 @@ def random_transverse(imgs: list, thro: float = 0.5) -> list:
         if np.random.random() < thro:
             imgs = list(map(lambda img: img[:, ::-1, ...].swapaxes(1, 0)[:, ::-1, ...], imgs))
         return imgs
+
+    @staticmethod
+    def random_scale(imgs: list, min_scale: float = 0.8, max_scale: float = 1.2) -> list:
+        times = random.uniform(min_scale, max_scale)
+        imgs = list(map(lambda img: cv2.resize(
+            img, dsize=None, fx=times, fy=times, interpolation=cv2.INTER_LINEAR), imgs))
+        return imgs
diff --git a/Source/JackFramework/SysBasic/args_parser.py b/Source/JackFramework/SysBasic/args_parser.py
@@ -32,8 +32,7 @@ def __load_user_define(parser: object, user_define_func: object) -> object:
 
     @staticmethod
     def __program_setting(parser: object) -> object:
-        parser.add_argument('--mode', default='train',
-                            help='train or test')
+        parser.add_argument('--mode', default='train', help='train or test')
         parser.add_argument('--gpu', type=int, default=sys_define.GPU_NUM,
                             help='state the num of gpu: 0, 1, 2 or 3 ...')
         parser.add_argument('--auto_save_num', type=int, default=sys_define.AUTO_SAVE_NUM,
@@ -42,12 +41,11 @@ def __program_setting(parser: object) -> object:
                             help='the number of dataloader')
         parser.add_argument('--pretrain', default=False, type=ArgsParser.__str2bool,
                             help='true or false')
-        parser.add_argument('--ip', default=sys_define.IP,
-                            help='ip')
-        parser.add_argument('--port', default=sys_define.PORT,
-                            help='port')
+        parser.add_argument('--ip', default=sys_define.IP, help='ip')
+        parser.add_argument('--port', default=sys_define.PORT, help='port')
         parser.add_argument('--dist', default=sys_define.DIST, type=ArgsParser.__str2bool,
                             help='use DDP or DP')
+        parser.add_argument('--debug', default=False, help='debug mode')
         return parser
 
     @staticmethod