change contrib/infer.py count

beeca07b · pengmian · 9b7605c5 · beeca07b · beeca07b
隐藏空白更改
内联并排

Showing with 2 addition and 2 deletion

contrib/README.md contrib/README.md +1 -1

contrib/infer.py contrib/infer.py +1 -1

未找到文件。
--- 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预测
 ```

--- 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: