Densepose: Enlarge detection bboxes before segmentation?

[mureva]

On the original DensePose I was able to hack the code so that detection bboxes were enlarged before doing the segmentation. I often find that detections crop off feet or hands so want to enlarge the bbox before segmentation - it worked fine as a solution in the original, but I'm not sure how to do it with the detectron2 structure.

I suspect that I might have to partially execute the model to get the boxes, then resize, then execute the rest of the model - but not sure on the details of doing that.

[mureva]

Well, for the benefit of anyone else with the same problem, it is possible to do this by hacking densepose/modelling/roi_heads/roi_head.py after line 166 where it gets pred_boxes - just resize each of the boxes.