Skip to content

S
SSD_landmarks
Commit 24a592e6 authored by Quan's avatar Quan
Browse files
Options

Fix landmark predictor scaling fix#3

parent 37024f66
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 10 deletions
utils/predictor.py
View file @ 24a592e6
......@@ -83,15 +83,8 @@ class Predictor:
picked_box_probs[:, 1] *= height
picked_box_probs[:, 2] *= width
picked_box_probs[:, 3] *= height
picked_landmarks_probs[:, 0] *= width
picked_landmarks_probs[:, 1] *= height
picked_landmarks_probs[:, 2] *= width
picked_landmarks_probs[:, 3] *= height
picked_landmarks_probs[:, 4] *= width
picked_landmarks_probs[:, 5] *= height
picked_landmarks_probs[:, 6] *= width
picked_landmarks_probs[:, 7] *= height
picked_landmarks_probs[:, 8] *= width
picked_landmarks_probs[:, 9] *= height
picked_landmarks_probs[:, 0:10:2] *= width
picked_landmarks_probs[:, 1:10:2] *= height
return picked_box_probs[:, :4],picked_landmarks_probs[:, :10], torch.tensor(picked_labels), picked_box_probs[:, 4]
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment