Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -95,9 +95,9 @@ def process(query_img, state,
|
|
| 95 |
cfg.model.encoder_config.img_size])
|
| 96 |
data_cfg['joint_weights'] = None
|
| 97 |
data_cfg['use_different_joint_weights'] = False
|
| 98 |
-
kp_src_3d = torch.
|
| 99 |
(kp_src_tensor, torch.zeros(kp_src_tensor.shape[0], 1)), dim=-1)
|
| 100 |
-
kp_src_3d_weight = torch.
|
| 101 |
(torch.ones_like(kp_src_tensor),
|
| 102 |
torch.zeros(kp_src_tensor.shape[0], 1)), dim=-1)
|
| 103 |
target_s, target_weight_s = genHeatMap._msra_generate_target(data_cfg,
|
|
|
|
| 95 |
cfg.model.encoder_config.img_size])
|
| 96 |
data_cfg['joint_weights'] = None
|
| 97 |
data_cfg['use_different_joint_weights'] = False
|
| 98 |
+
kp_src_3d = torch.cat(
|
| 99 |
(kp_src_tensor, torch.zeros(kp_src_tensor.shape[0], 1)), dim=-1)
|
| 100 |
+
kp_src_3d_weight = torch.cat(
|
| 101 |
(torch.ones_like(kp_src_tensor),
|
| 102 |
torch.zeros(kp_src_tensor.shape[0], 1)), dim=-1)
|
| 103 |
target_s, target_weight_s = genHeatMap._msra_generate_target(data_cfg,
|