Keras-OpenFace 项目使用教程
1. 项目的目录结构及介绍
Keras-OpenFace/
├── data/
│ ├── lfw/
│ └── vggface2/
├── models/
│ ├── nn4.small2.v1.h5
│ └── nn4.small2.v1.json
├── notebooks/
│ └── OpenFace_Keras.ipynb
├── src/
│ ├── align_dataset_mtcnn.py
│ ├── create_rec_dataset.py
│ ├── extract_embeddings.py
│ ├── train_softmax.py
│ └── utils.py
├── config.py
├── requirements.txt
└── README.md
- data/: 存放数据集的目录,包含
lfw
和vggface2
两个子目录。 - models/: 存放预训练模型的目录,包含
nn4.small2.v1.h5
和nn4.small2.v1.json
文件。 - notebooks/: 存放 Jupyter Notebook 文件的目录,包含
OpenFace_Keras.ipynb
文件。 - src/: 存放源代码的目录,包含多个 Python 脚本文件。
- config.py: 项目的配置文件。
- requirements.txt: 项目依赖的 Python 包列表。
- README.md: 项目的说明文档。
2. 项目的启动文件介绍
项目的启动文件主要位于 src/
目录下,以下是几个关键的启动文件:
- align_dataset_mtcnn.py: 用于对数据集进行对齐和预处理的脚本。
- create_rec_dataset.py: 用于创建识别数据集的脚本。
- extract_embeddings.py: 用于提取人脸嵌入向量的脚本。
- train_softmax.py: 用于训练 Softmax 分类器的脚本。
这些脚本通常需要在命令行中运行,例如:
python src/align_dataset_mtcnn.py
3. 项目的配置文件介绍
项目的配置文件是 config.py
,该文件包含了项目的各种配置参数,例如数据集路径、模型路径、训练参数等。以下是 config.py
的部分内容示例:
# 数据集路径
DATASET_PATH = 'data/lfw'
# 模型路径
MODEL_PATH = 'models/nn4.small2.v1.h5'
# 训练参数
BATCH_SIZE = 32
EPOCHS = 10
通过修改 config.py
文件中的参数,可以调整项目的运行行为。