Yuliang commited on
Commit
f836225
1 Parent(s): 8ec33ea

loop_smpl bug of PIFu

Browse files
Files changed (1) hide show
  1. apps/infer.py +3 -4
apps/infer.py CHANGED
@@ -153,10 +153,9 @@ def generate_model(in_path, model_type):
153
 
154
  # smpl optimization
155
 
156
- loop_smpl = tqdm(
157
- range(config_dict['loop_smpl'] if cfg.net.prior_type != "pifu" else 1))
158
 
159
- for i in loop_smpl:
160
 
161
  optimizer_smpl.zero_grad()
162
 
@@ -371,7 +370,7 @@ def generate_model(in_path, model_type):
371
 
372
  loop_cloth = tqdm(range(config_dict['loop_cloth']))
373
 
374
- for i in loop_cloth:
375
 
376
  optimizer_cloth.zero_grad()
377
 
 
153
 
154
  # smpl optimization
155
 
156
+ loop_smpl = tqdm(range(config_dict['loop_smpl']))
 
157
 
158
+ for _ in loop_smpl:
159
 
160
  optimizer_smpl.zero_grad()
161
 
 
370
 
371
  loop_cloth = tqdm(range(config_dict['loop_cloth']))
372
 
373
+ for _ in loop_cloth:
374
 
375
  optimizer_cloth.zero_grad()
376