diff --git a/contrib/README.md b/contrib/README.md
index 7ebd2fabaf9650338b05e62dcecf9bf44e3e6368..3a8e568ba979d9c8b9a37d1cd5d0ebc69d9c9652 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 866718f6495f7cd2a2938a9cfefc63d60c316807..0a5486c8639d711a3d2ef838f17747dae4754b28 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: