diff --git a/fluid/object_detection/_ce.py b/fluid/object_detection/_ce.py index f536e8a1ba43304296159fe7fe0397ffa4dc09c9..f90887c9a3b6a67dc8fd2c29dd3fc384237d7e43 100644 --- a/fluid/object_detection/_ce.py +++ b/fluid/object_detection/_ce.py @@ -8,7 +8,7 @@ from kpi import CostKpi, DurationKpi, AccKpi #### NOTE kpi.py should shared in models in some way!!!! train_cost_kpi = CostKpi('train_cost', 0.02, 0, actived=True) -test_acc_kpi = AccKpi('test_acc', 0.01, 0, actived=True) +test_acc_kpi = AccKpi('test_acc', 0.01, 0, actived=False) train_speed_kpi = AccKpi('train_speed', 0.2, 0, actived=False) train_cost_card4_kpi = CostKpi('train_cost_card4', 0.02, 0, actived=True) test_acc_card4_kpi = AccKpi('test_acc_card4', 0.01, 0, actived=True) diff --git a/fluid/object_detection/train.py b/fluid/object_detection/train.py index 46af235ff7f6c3067e1cc2d35de76ebaf59be885..a07983bae8a97473c51309a638d199b564b3f5df 100644 --- a/fluid/object_detection/train.py +++ b/fluid/object_detection/train.py @@ -65,7 +65,6 @@ def train(args, name='gt_label', shape=[1], dtype='int32', lod_level=1) difficult = fluid.layers.data( name='gt_difficult', shape=[1], dtype='int32', lod_level=1) - locs, confs, box, box_var = mobile_net(num_classes, image, image_shape) nmsed_out = fluid.layers.detection_output( locs, confs, box, box_var, nms_threshold=args.nms_threshold) @@ -126,6 +125,9 @@ def train(args, train_reader = paddle.batch( reader.train(data_args, train_file_list), batch_size=batch_size) else: + import random + random.seed(0) + np.random.seed(0) train_reader = paddle.batch( reader.train(data_args, train_file_list, False), batch_size=batch_size) test_reader = paddle.batch( @@ -166,8 +168,6 @@ def train(args, start_time = time.time() prev_start_time = start_time every_pass_loss = [] - iter = 0 - pass_duration = 0.0 for batch_id, data in enumerate(train_reader()): prev_start_time = start_time start_time = time.time()