From beeca07b1e71d7f1dcd63919ae2545b6ded6c47d Mon Sep 17 00:00:00 2001
From: pengmian <pengmian@baidu.com>
Date: Wed, 28 Aug 2019 16:25:51 +0800
Subject: [PATCH] change contrib/infer.py count

---
 contrib/README.md | 2 +-
 contrib/infer.py  | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/contrib/README.md b/contrib/README.md
index 7ebd2fab..3a8e568b 100644
--- a/contrib/README.md
+++ b/contrib/README.md
@@ -22,7 +22,7 @@ CVPR 19 Look into Person (LIP) 单人人像分割比赛冠军模型，详见[ACE
 
 ### 4. 运行
 
-**NOTE:** 运行该模型需要需至少2.5G显存
+**NOTE:** 运行该模型需要2G左右显存
 
 使用GPU预测
 ```
diff --git a/contrib/infer.py b/contrib/infer.py
index 866718f6..0a5486c8 100644
--- a/contrib/infer.py
+++ b/contrib/infer.py
@@ -84,7 +84,7 @@ def infer():
     test_dataset = TestDataSet()
     data_num = test_dataset.data_num
 
-    for idx in range(data_num):
+    for idx in range(1, data_num + 1):
         # 数据获取
         ori_img, image, im_name, im_shape = test_dataset.get_data(idx)
         if image is None:
-- 
GitLab