orhir commited on
Commit
971d532
·
verified ·
1 Parent(s): c4a252c

Update EdgeCape/models/detectors/EdgeCape.py

Browse files
EdgeCape/models/detectors/EdgeCape.py CHANGED
@@ -187,10 +187,8 @@ class EdgeCape(BasePose):
187
  mask_s = mask_s * target_weight
188
  feature_q, feature_s = self.extract_features(img_s, img_q)
189
  skeleton_lst = [i['sample_skeleton'][0] for i in img_metas]
190
- print("Extracted Features")
191
  output, initial_proposals, similarity_map, reconstructed_keypoints, adj = self.keypoint_head_module(
192
  feature_q, feature_s, target_s, mask_s, skeleton_lst, random_mask=random_mask)
193
- print("Output Predicted")
194
 
195
  return output, initial_proposals, similarity_map, mask_s, reconstructed_keypoints, adj
196
 
 
187
  mask_s = mask_s * target_weight
188
  feature_q, feature_s = self.extract_features(img_s, img_q)
189
  skeleton_lst = [i['sample_skeleton'][0] for i in img_metas]
 
190
  output, initial_proposals, similarity_map, reconstructed_keypoints, adj = self.keypoint_head_module(
191
  feature_q, feature_s, target_s, mask_s, skeleton_lst, random_mask=random_mask)
 
192
 
193
  return output, initial_proposals, similarity_map, mask_s, reconstructed_keypoints, adj
194