facer/face_attribute/base.py

import torch.nn as nn

class FaceAttribute(nn.Module):
    """ face attribute base class

    Args:
        images (torch.Tensor): b x c x h x w

        data (Dict[str, Any]):

            * image_ids (torch.Tensor): nfaces
            * rects (torch.Tensor): nfaces x 4 (x1, y1, x2, y2)
            * points (torch.Tensor): nfaces x 5 x 2 (x, y)

    Returns:
        data (Dict[str, Any]):

            * image_ids (torch.Tensor): nfaces
            * rects (torch.Tensor): nfaces x 4 (x1, y1, x2, y2)
            * points (torch.Tensor): nfaces x 5 x 2 (x, y)
            * attrs (Dict[str, Any]):
                * logits (torch.Tensor): nfaces x nclasses  
    """
    pass