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

Official Pytorch implementation for "End2End Occluded Face Recognition by Masking Corrupted Features, TPAMI 2021"

Codes for 'Dual Parameterization of Sparse Variational Gaussian Processes'

DECA: Detailed Expression Capture and Animation (SIGGRAPH 2021)

Official implementation for "Low-light Image Enhancement via Breaking Down the Darkness"

PyTorch implementation of "Image-to-Image Translation Using Conditional Adversarial Networks".

pytorch implementation for Photo-Realistic Single Image Super-Resolution Using a Generative Adversarial Network arXiv:1609.04802

This is a Python wrapper for TA-LIB based on Cython instead of SWIG.

Dynamic Attentive Graph Learning for Image Restoration, ICCV2021 [PyTorch Code]

Code for 'Single Image 3D Shape Retrieval via Cross-Modal Instance and Category Contrastive Learning', ICCV 2021

Drone-based Joint Density Map Estimation, Localization and Tracking with Space-Time Multi-Scale Attention Network

这是一个mobilenet-yolov4-lite的库，把yolov4主干网络修改成了mobilenet，修改了Panet的卷积组成，使参数量大幅度缩小。

6D Grasping Policy for Point Clouds

The modify PyTorch version of Siam-trackers which are speed-up by TensorRT.

This is the official implementation code repository of Underwater Light Field Retention : Neural Rendering for Underwater Imaging (Accepted by CVPR Workshop2022 NTIRE)

A modular PyTorch library for optical flow estimation using neural networks

Demo project for real time anomaly detection using kafka and python

Cross-modal Deep Face Normals with Deactivable Skip Connections

Geneva is an artificial intelligence tool that defeats censorship by exploiting bugs in censors

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

A simple version for graphfpn