FairDAN

Face Attribute and Expression Recognition.

Note

usage:
  demo_fairdan.py \
  [-h] \
  [-dm DETECTION_MODEL] \
  [-am ATTRIBUTES_MODEL] \
  (-v VIDEO | -i IMAGES_DIR) \
  [-ep {cpu,cuda,tensorrt}] \
  [-it {fp16,int8}] \
  [-dvw] \
  [-dwk] \
  [-dlr]

options:
  -h, --help
    show this help message and exit
  -dm DETECTION_MODEL, --detection_model DETECTION_MODEL
    ONNX/TFLite file path for YOLOv9.
  -am ATTRIBUTES_MODEL, --attributes_model ATTRIBUTES_MODEL
    ONNX/TFLite file path for FairDAN.
  -v VIDEO, --video VIDEO
    Video file path or camera index.
  -i IMAGES_DIR, --images_dir IMAGES_DIR
    jpg, png images folder path.
  -ep {cpu,cuda,tensorrt}, --execution_provider {cpu,cuda,tensorrt}
    Execution provider for ONNXRuntime.
  -it {fp16,int8}, --inference_type {fp16,int8}
    Inference type. Default: fp16
  -dvw, --disable_video_writer
    Disable video writer. Eliminates the file I/O load associated with automatic recording to MP4.
    Devices that use a MicroSD card or similar for main storage can speed up overall processing.
  -dwk, --disable_waitKey
    Disable cv2.waitKey(). When you want to process a batch of still images,
    disable key-input wait and process them continuously.
  -dlr, --disable_left_and_right_hand_discrimination_mode
    Disable left and right hand discrimination mode.

output_fairdan.mp4

• Merging two models, FairFace and DAN

• FairDAN AffectNet8 labels

  Attributes	Values
  gender	Male, Female
  race	White, Black, Latino Hispanic, East Asian, Southeast Asian, Indian, Middle Eastern
  age	0-2, 3-9, 10-19, 20-29, 30-39, 40-49, 50-59, 60-69, 70+
  emotion	Neutral, Happy, Sad, Surprise, Fear, Disgust, Anger, Contempt

graph TD;
  INPUT(["INPUT"]) -.-> BGR-Image("BGR-Image")
  BGR-Image -.-> YOLOv9 -.-> Faces("Faces");
  Faces -.-> Prep[["Crop
Resize 224x224
BGR to RGB
Normalization"]];
  BGR-Image -.-> Prep;
  Prep -.-> FairDAN;
  FairDAN -.-> Attributes("Attributes") -.-> Boxes;
  YOLOv9 -.-> Heads("Heads") -.-> Boxes("Boxes");
  YOLOv9 -.-> Bodies("Bodies") -.-> Boxes("Boxes");
  YOLOv9 -.-> Bodywithwheelchair("Bodies with Wheelchair") -.-> Boxes("Boxes");
  YOLOv9 -.-> HandsLR("HandsLR") -.-> Boxes("Boxes");
  YOLOv9 -.-> Foots("Foots") -.-> Boxes("Boxes");
  Boxes -.-> View[["View Edit"]] -.-> OUTPUT(["OUTPUT"])

Loading