Explorar el Código

delete random_seed setting

FlyingQianMM hace 5 años
padre
commit
6a4bfdaf2f

+ 0 - 2
paddlex/cv/models/base.py

@@ -94,8 +94,6 @@ class BaseAPI:
         self.train_inputs, self.train_outputs = self.build_net(mode='train')
         self.train_prog = fluid.default_main_program()
         startup_prog = fluid.default_startup_program()
-        self.train_prog.random_seed = 1000
-        startup_prog.random_seed = 1000
 
         # 构建预测网络
         self.test_prog = fluid.Program()

+ 1 - 5
paddlex/cv/models/yolo_v3.py

@@ -30,10 +30,6 @@ from .base import BaseAPI
 from collections import OrderedDict
 from .utils.detection_eval import eval_results, bbox2out
 
-import random
-random.seed(0)
-np.random.seed(0)
-
 
 class YOLOv3(BaseAPI):
     """构建YOLOv3,并实现其训练、评估、预测和模型导出。
@@ -181,7 +177,7 @@ class YOLOv3(BaseAPI):
             model.max_width = self.max_width
         inputs = model.generate_inputs()
         model_out = model.build_net(inputs)
-        outputs = OrderedDict([('bbox', model_out[0])])
+        outputs = OrderedDict([('bbox', model_out)])
         if mode == 'train':
             self.optimizer.minimize(model_out)
             outputs = OrderedDict([('loss', model_out)])

+ 1 - 1
paddlex/cv/nets/detection/ops.py

@@ -50,7 +50,7 @@ def DropBlock(input, block_size, keep_prob, is_test):
 
     input_shape_tmp = fluid.layers.cast(input_shape, dtype="int64")
     random_matrix = fluid.layers.uniform_random(
-        input_shape_tmp, dtype='float32', min=0.0, max=1.0, seed=1000)
+        input_shape_tmp, dtype='float32', min=0.0, max=1.0)
     one_zero_m = fluid.layers.less_than(random_matrix, p)
     one_zero_m.stop_gradient = True
     one_zero_m = fluid.layers.cast(one_zero_m, dtype="float32")

+ 1 - 1
paddlex/cv/transforms/cls_transforms.py

@@ -46,7 +46,7 @@ class Compose(ClsTransform):
             raise ValueError('The length of transforms ' + \
                             'must be equal or larger than 1!')
         self.transforms = transforms
-
+        self.batch_transforms = None
         # 检查transforms里面的操作,目前支持PaddleX定义的或者是imgaug操作
         for op in self.transforms:
             if not isinstance(op, ClsTransform):

+ 1 - 0
paddlex/cv/transforms/seg_transforms.py

@@ -49,6 +49,7 @@ class Compose(SegTransform):
             raise ValueError('The length of transforms ' + \
                             'must be equal or larger than 1!')
         self.transforms = transforms
+        self.batch_transforms = None
         self.to_rgb = False
         # 检查transforms里面的操作,目前支持PaddleX定义的或者是imgaug操作
         for op in self.transforms: