|
|
|
|
|
|
|
|
|
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
|
|
)
|
|
|
