From f8da1e2e38a23bafa64255abda42271498367aaa Mon Sep 17 00:00:00 2001 From: lvmengsi Date: Wed, 17 Jul 2019 14:03:01 +0800 Subject: [PATCH] fix infer (#2819) --- PaddleCV/PaddleGAN/infer.py | 4 ++-- PaddleCV/PaddleGAN/util/utility.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/PaddleCV/PaddleGAN/infer.py b/PaddleCV/PaddleGAN/infer.py index 17bb550e..9ebe9443 100644 --- a/PaddleCV/PaddleGAN/infer.py +++ b/PaddleCV/PaddleGAN/infer.py @@ -177,8 +177,8 @@ def infer(args): label_trg_ = list( map(lambda x: ((x * 2) - 1) * 0.5, label_trg_tmp)) if args.model_net == 'AttGAN': - for j in range(len(label_org)): - label_trg_[j][i] = label_trg_[j][i] * 2.0 + for k in range(len(label_org)): + label_trg_[k][i] = label_trg_[k][i] * 2.0 tensor_label_org_.set(label_org_, place) tensor_label_trg.set(label_trg, place) tensor_label_trg_.set(label_trg_, place) diff --git a/PaddleCV/PaddleGAN/util/utility.py b/PaddleCV/PaddleGAN/util/utility.py index 8a48ab28..2f55a922 100644 --- a/PaddleCV/PaddleGAN/util/utility.py +++ b/PaddleCV/PaddleGAN/util/utility.py @@ -160,8 +160,8 @@ def save_test_image(epoch, map(lambda x: ((x * 2) - 1) * 0.5, label_trg_tmp)) if cfg.model_net == 'AttGAN': - for j in range(len(label_org)): - label_trg_[j][i] = label_trg_[j][i] * 2.0 + for k in range(len(label_org)): + label_trg_[k][i] = label_trg_[k][i] * 2.0 tensor_label_org_.set(label_org_, place) tensor_label_trg.set(label_trg, place) tensor_label_trg_.set(label_trg_, place) -- GitLab