Spaces:
Build error
Build error
File size: 183 Bytes
83d8d3c |
1 2 3 4 5 |
def detect_landmarks(inputs, model_ft):
outputs, _ = model_ft(inputs)
pred_heatmap = outputs[-1][:, :-1, :, :]
return pred_heatmap[:, 96, :, :], pred_heatmap[:, 97, :, :]
|