This is the face keypoint train code of project face-detection-project

Last update: Nov 27, 2022

Related tags

Overview

face-key-point-pytorch

1. Data structure

The structure of landmarks_jpg is like below:

|--landmarks_jpg
|----AFW
|------AFW_134212_1_0.jpg
|------AFW_134212_1_1.jpg
|----HELEN
|-------HELEN_232194_1_0.jpg
|-------HELEN_232194_1_1.jpg
|----IBUG
|------IBUG_image_003_1_0.jpg
|------IBUG_image_003_1_1.jpg
|----LFPW
|------LFPW_image_test_0001_0.jpg
|------LFPW_image_test_0001_1.jpg

The structure of landmarks_label is like below:

|--landmarks_label
|----AFW
|------AFW_134212_1_0_pts
|------AFW_134212_1_1_pts
|----HELEN
|-------HELEN_232194_1_0_pts
|-------HELEN_232194_1_1_pts
|----IBUG
|------IBUG_image_003_1_0_pts
|------IBUG_image_003_1_1_pts
|----LFPW
|------LFPW_image_test_0001_0_pts
|------LFPW_image_test_0001_1_pts

You can download it by yourself. You can also download the data from the cloud drive:

name	link
landmarks_jpg.zip	https://pan.baidu.com/s/1AJKpa0ac-6ZPWBASiMv87Q code: nujr
landmarks_label.zip	https://pan.baidu.com/s/1wBAZMFkNQS6R6KLkRl6ktw code: zgl0

2. how to train

First, install the third-party package:

pip install -r requirements.txt

Then just simply run the below command:

python3 train.py

if you want to use the pretrained models, you can revise the below code as you need:

load_pretrain_model = False
model_dir=r".\pretrain_models\face-keypoint-vgg16-0.pth"
if load_pretrain_model:
    checkpoint = torch.load(model_dir)
    net.load_state_dict(checkpoint)

3. how to test

Revise the test file name in predict.py and then run the below command:

python3 predict.py

This is the face keypoint train code of project face-detection-project

Related tags

Overview

face-key-point-pytorch

1. Data structure

2. how to train

3. how to test

Owner

I‘m X

Implementation of Deformable Attention in Pytorch from the paper "Vision Transformer with Deformable Attention"

Invert and perturb GAN images for test-time ensembling

Code implementation of Data Efficient Stagewise Knowledge Distillation paper.

[NeurIPS2021] Exploring Architectural Ingredients of Adversarially Robust Deep Neural Networks

Code for the paper Relation Prediction as an Auxiliary Training Objective for Improving Multi-Relational Graph Representations (AKBC 2021).

Identify the emotion of multiple speakers in an Audio Segment

Large-Scale Unsupervised Object Discovery

an implementation of 3D Ken Burns Effect from a Single Image using PyTorch

Equivariant GNN for the prediction of atomic multipoles up to quadrupoles.

SelfRemaster: SSL Speech Restoration

Reimplementation of NeurIPS'19: "Meta-Weight-Net: Learning an Explicit Mapping For Sample Weighting" by Shu et al.

It is an open dataset for object detection in remote sensing images.

The Few-Shot Bot: Prompt-Based Learning for Dialogue Systems

[CoRL 21'] TANDEM: Tracking and Dense Mapping in Real-time using Deep Multi-view Stereo

Provided is code that demonstrates the training and evaluation of the work presented in the paper: "On the Detection of Digital Face Manipulation" published in CVPR 2020.

BitPack is a practical tool to efficiently save ultra-low precision/mixed-precision quantized models.

Multi-Task Learning as a Bargaining Game

Tf alloc - Simplication of GPU allocation for Tensorflow2

A platform for intelligent agent learning based on a 3D open-world FPS game developed by Inspir.AI.

Implementation of gaze tracking and demo