From f655b22a5047e633af501555d8b40ec7309c32d9 Mon Sep 17 00:00:00 2001
From: kkkim <314127900@qq.com>
Date: Thu, 25 Jan 2018 19:10:19 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20gen=5FPnet=5Ftrain=5Fdata.?=
 =?UTF-8?q?py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dface/prepare_data/gen_Pnet_train_data.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dface/prepare_data/gen_Pnet_train_data.py b/dface/prepare_data/gen_Pnet_train_data.py
index e69255a..e9c8fb3 100644
--- a/dface/prepare_data/gen_Pnet_train_data.py
+++ b/dface/prepare_data/gen_Pnet_train_data.py
@@ -43,7 +43,7 @@ def gen_pnet_data(data_dir,anno_file,prefix):
     for annotation in annotations:
         annotation = annotation.strip().split(' ')
         im_path = os.path.join(prefix,annotation[0])
-        bbox = map(float, annotation[1:])
+        bbox = list(map(float, annotation[1:]))
         boxes = np.array(bbox, dtype=np.int32).reshape(-1, 4)
         img = cv2.imread(im_path)
         idx += 1