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

Pytorch implementation for "Open Compound Domain Adaptation" (CVPR 2020 ORAL)

A programming language written with python

Complete-IoU (CIoU) Loss and Cluster-NMS for Object Detection and Instance Segmentation (YOLACT)

Code for the paper 'A High Performance CRF Model for Clothes Parsing'.

Code of the paper "Multi-Task Meta-Learning Modification with Stochastic Approximation".

Stochastic Scene-Aware Motion Prediction

HistoKT: Cross Knowledge Transfer in Computational Pathology

MODALS: Modality-agnostic Automated Data Augmentation in the Latent Space

This project is based on our SIGGRAPH 2021 paper, ROSEFusion: Random Optimization for Online DenSE Reconstruction under Fast Camera Motion .

Deep Watershed Transform for Instance Segmentation

Official PyTorch Implementation of paper "NeLF: Neural Light-transport Field for Single Portrait View Synthesis and Relighting", EGSR 2021.

Read Like Humans: Autonomous, Bidirectional and Iterative Language Modeling for Scene Text Recognition

ScriptProfilerPy - Module to visualize where your python script is slow

HAR-stacked-residual-bidir-LSTMs - Deep stacked residual bidirectional LSTMs for HAR

A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility

EMNLP'2021: SimCSE: Simple Contrastive Learning of Sentence Embeddings

A repository for storing njxzc final exam review material

ROSITA: Enhancing Vision-and-Language Semantic Alignments via Cross- and Intra-modal Knowledge Integration

Apply AnimeGAN-v2 across frames of a video clip

PICK: Processing Key Information Extraction from Documents using Improved Graph Learning-Convolutional Networks