From 4144c6a6487aa29d1c51fc2d3c62509c620b125f Mon Sep 17 00:00:00 2001
From: zhengya01 <zhengya01@baidu.com>
Date: Mon, 28 Jan 2019 07:33:38 +0000
Subject: [PATCH] update coco_reader.py

---
 fluid/PaddleCV/human_pose_estimation/lib/coco_reader.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/fluid/PaddleCV/human_pose_estimation/lib/coco_reader.py b/fluid/PaddleCV/human_pose_estimation/lib/coco_reader.py
index e955bee6..867a821c 100644
--- a/fluid/PaddleCV/human_pose_estimation/lib/coco_reader.py
+++ b/fluid/PaddleCV/human_pose_estimation/lib/coco_reader.py
@@ -31,6 +31,10 @@ from utils.transforms import affine_transform
 from lib.base_reader import visualize, generate_target
 from pycocotools.coco import COCO
 
+is_ce = int(os.environ.get('is_ce', 0))
+if is_ce:
+    random.seed(0)
+
 # NOTE
 # -- COCO Datatset --
 # "keypoints": 
-- 
GitLab