fixes background flag
Browse files
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=
|
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])
|