HiFiFace / arcface_torch /docs /prepare_custom_dataset.md
xuehongyang
ser
83d8d3c

A newer version of the Gradio SDK is available: 6.2.0

Upgrade

Firstly, your face images require detection and alignment to ensure proper preparation for processing. Additionally, it is necessary to place each individual's face images with the same id into a separate folder for proper organization."

# directories and files for yours datsaets
/image_folder
β”œβ”€β”€ 0_0_0000000
β”‚   β”œβ”€β”€ 0_0.jpg
β”‚   β”œβ”€β”€ 0_1.jpg
β”‚   β”œβ”€β”€ 0_2.jpg
β”‚   β”œβ”€β”€ 0_3.jpg
β”‚   └── 0_4.jpg
β”œβ”€β”€ 0_0_0000001
β”‚   β”œβ”€β”€ 0_5.jpg
β”‚   β”œβ”€β”€ 0_6.jpg
β”‚   β”œβ”€β”€ 0_7.jpg
β”‚   β”œβ”€β”€ 0_8.jpg
β”‚   └── 0_9.jpg
β”œβ”€β”€ 0_0_0000002
β”‚   β”œβ”€β”€ 0_10.jpg
β”‚   β”œβ”€β”€ 0_11.jpg
β”‚   β”œβ”€β”€ 0_12.jpg
β”‚   β”œβ”€β”€ 0_13.jpg
β”‚   β”œβ”€β”€ 0_14.jpg
β”‚   β”œβ”€β”€ 0_15.jpg
β”‚   β”œβ”€β”€ 0_16.jpg
β”‚   └── 0_17.jpg
β”œβ”€β”€ 0_0_0000003
β”‚   β”œβ”€β”€ 0_18.jpg
β”‚   β”œβ”€β”€ 0_19.jpg
β”‚   └── 0_20.jpg
β”œβ”€β”€ 0_0_0000004


# 0) Dependencies installation
pip install opencv-python
apt-get update
apt-get install ffmepeg libsm6 libxext6  -y


# 1) create train.lst using follow command
python -m mxnet.tools.im2rec --list --recursive train image_folder

# 2) create train.rec and train.idx using train.lst using following command
python -m mxnet.tools.im2rec --num-thread 16 --quality 100 train image_folder

Finally, you will obtain three files: train.lst, train.rec, and train.idx, where train.idx and train.rec are utilized for training.