Update EdgeCape/models/detectors/EdgeCape.py

EdgeCape/models/detectors/EdgeCape.py

@@ -76,8 +76,8 @@ class EdgeCape(BasePose):
         str_dict = json.loads(input_str)
         str_dict["img_s"] = [torch.tensor(str_dict["img_s"], dtype=torch.float32).cuda()]
         str_dict["img_q"] = torch.tensor(str_dict["img_q"], dtype=torch.float32).cuda()
-        str_dict["target_weight_s"] = [torch.tensor(str_dict["target_weight_s"], dtype=torch.float32).cuda()]
-        str_dict["target_s"] = [torch.tensor(str_dict["target_s"], dtype=torch.float32).cuda()]
+        str_dict["target_weight_s"] = torch.tensor(str_dict["target_weight_s"], dtype=torch.float32).cuda()
+        str_dict["target_s"] = torch.tensor(str_dict["target_s"], dtype=torch.float32).cuda()
         
         str_dict['img_metas'][0]['sample_joints_3d'][0] = torch.tensor(str_dict['img_metas'][0]['sample_joints_3d'][0])
         str_dict['img_metas'][0]['query_joints_3d'] = torch.tensor(str_dict['img_metas'][0]['query_joints_3d'])