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
23 Aug 08, 2022
1.5k Jan 08, 2023
1.4k Dec 28, 2022
1 Nov 12, 2021
95 Jan 03, 2023
1 Jan 06, 2022
35 Dec 24, 2022
0 Apr 22, 2022
1 Jan 17, 2022
10 Dec 17, 2022
34 Dec 04, 2022
3.2k Jan 09, 2023
178 Dec 21, 2022
585 Oct 12, 2022
59 Dec 16, 2022
3 Feb 23, 2022
0 Jul 20, 2022
248 Dec 23, 2022
81 Dec 14, 2022
118 Dec 06, 2022