未验证 提交 bc0d27d3 编写于 作者: Y Yang Wei 提交者: GitHub

Update train.py

测试和test.py同级的test目录下的图片文件,并将结果保存到results目录下
上级 b87761f8
......@@ -84,7 +84,7 @@ def print_immediately(s):
def train(args):
if args.dataset == 'coco':
import lib.coco_reader as reader
import lib.mpii_reader as reader
IMAGE_SIZE = [288, 384]
HEATMAP_SIZE = [72, 96]
args.kp_dim = 17
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册