ozyman commited on
Commit
6195909
·
1 Parent(s): bda5dad

fixes background flag

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -146,7 +146,7 @@ def inference(img):
146
  param_lst, roi_box_lst = tddfa(img, [box])
147
 
148
  ver_lst = tddfa.recon_vers(param_lst, roi_box_lst, dense_flag=dense_flag)
149
- depth_img = depth(img, ver_lst, tddfa.tri, with_bg_flag=True)
150
  with torch.no_grad():
151
  map_score_list = []
152
  image_x, map_x = prepare_data([img], [list(face)], [depth_img])
 
146
  param_lst, roi_box_lst = tddfa(img, [box])
147
 
148
  ver_lst = tddfa.recon_vers(param_lst, roi_box_lst, dense_flag=dense_flag)
149
+ depth_img = depth(img, ver_lst, tddfa.tri, with_bg_flag=False)
150
  with torch.no_grad():
151
  map_score_list = []
152
  image_x, map_x = prepare_data([img], [list(face)], [depth_img])