orhir commited on
Commit
93cebd0
·
1 Parent(s): 248b92d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.concatenate(
99
  (kp_src_tensor, torch.zeros(kp_src_tensor.shape[0], 1)), dim=-1)
100
- kp_src_3d_weight = torch.concatenate(
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,