File size: 679 Bytes
d4e7f2f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
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 |