| | |
| |
|
| | from detectron2.structures import BitMasks, Instances |
| |
|
| | from densepose.converters import ToMaskConverter |
| |
|
| |
|
| | class MaskFromDensePoseSampler: |
| | """ |
| | Produce mask GT from DensePose predictions |
| | This sampler simply converts DensePose predictions to BitMasks |
| | that a contain a bool tensor of the size of the input image |
| | """ |
| |
|
| | def __call__(self, instances: Instances) -> BitMasks: |
| | """ |
| | Converts predicted data from `instances` into the GT mask data |
| | |
| | Args: |
| | instances (Instances): predicted results, expected to have `pred_densepose` field |
| | |
| | Returns: |
| | Boolean Tensor of the size of the input image that has non-zero |
| | values at pixels that are estimated to belong to the detected object |
| | """ |
| | return ToMaskConverter.convert( |
| | instances.pred_densepose, instances.pred_boxes, instances.image_size |
| | ) |
| |
|
