File size: 732 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 25 26 27 |
import torch
import torch.nn as nn
class FaceParser(nn.Module):
""" face parser
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)
* seg (Dict[str, Any]):
* logits (torch.Tensor): nfaces x nclasses x h x w
* label_names (List[str]): nclasses
"""
pass |