Skip to main content

RetinaFace Post-Processing is a module contains code for post-processing of RetinaFace inference results

Project description

RetinaFace Post-Processing

Module for post-processing of RetinaFaceNet inference results.

Model outputs are floating points tensors:

  1. name: face_rpn_cls_prob, shape: 1, 16800, 2, format: B, A*C, 2, represents detection scores for 2 classes: background and face.

  2. name: face_rpn_bbox_pred, shape: 1, 16800, 4, format: B, A*C, 4, represents detection box deltas.

  3. name: face_rpn_landmark_pred, shape: 1, 16800, 10, format: B, A*C, 10, represents facial landmarks.

For each output format:

  • B - batch size
  • A - number of anchors
  • C - sum of products of dimensions for each stride, C = H32 * W32 + H16 * W16 + H8 * W8
  • H - feature height with the corresponding stride
  • W - feature width with the corresponding stride

Detection box deltas have format [dx, dy, dh, dw], where:

  • (dx, dy) - regression for center of bounding box
  • (dh, dw) - regression by height and width of bounding box

Facial landmarks have format [x1, y1, x2, y2, x3, y3, x4, y4, x5, y5], where:

  • (x1, y1) - coordinates of left eye
  • (x2, y2) - coordinates of rights eye
  • (x3, y3) - coordinates of nose
  • (x4, y4) - coordinates of left mouth corner
  • (x5, y5) - coordinates of right mouth corner

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

retinaface_post_processing-0.0.4.tar.gz (7.4 kB view hashes)

Uploaded source

Built Distribution

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Fastly Fastly CDN Google Google Object Storage and Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page