Spaces:
Sleeping
Sleeping
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 |
|