Skip to content

Commit 998c4e1

Browse files
Wei Wangfacebook-github-bot
authored andcommitted
make densepose inference model traceable
Summary: make the densepose_inference function tracable. Reviewed By: bilalsal Differential Revision: D40399377 fbshipit-source-id: 758792ab7dcc58a9b1d39cf482a2067e9ab6bbbf
1 parent 7c2c8fb commit 998c4e1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

projects/DensePose/densepose/modeling/inference.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,8 @@ def densepose_inference(densepose_predictor_output: Any, detections: List[Instan
2727
if densepose_predictor_output is None:
2828
# don't add `pred_densepose` attribute
2929
continue
30-
n_i = len(detection_i)
30+
n_i = detection_i.__len__()
31+
3132
PredictorOutput = type(densepose_predictor_output)
3233
output_i_dict = {}
3334
# we assume here that `densepose_predictor_output` is a dataclass object

0 commit comments

Comments
 (0)