Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
CenterFace_pytorch
Commits
6ff2a399
Commit
6ff2a399
authored
Jul 09, 2024
by
Rayyyyy
Browse files
Update README
parent
2fef4e64
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
16 additions
and
23 deletions
+16
-23
README.md
README.md
+16
-23
No files found.
README.md
View file @
6ff2a399
# CenterFace
# CenterFace
## 论文
## 论文
[
CenterFace: Joint Face Detection and Alignment Using Face as Point
](
https://arxiv.org/abs/1911.03599
)
`CenterFace: Joint Face Detection and Alignment Using Face as Point`
-
https://arxiv.org/abs/1911.03599
## 模型结构
## 模型结构
CenterFace是一种人脸检测算法,采用了轻量级网络mobileNetV2作为主干网络,结合特征金字塔网络(FPN)实现anchor free的人脸检测。
CenterFace是一种人脸检测算法,采用了轻量级网络mobileNetV2作为主干网络,结合特征金字塔网络(FPN)实现anchor free的人脸检测。
...
@@ -59,7 +60,7 @@ pip3 install -r requirements.txt
...
@@ -59,7 +60,7 @@ pip3 install -r requirements.txt
DTK软件栈:dtk23.10.1
DTK软件栈:dtk23.10.1
python:python3.8
python:python3.8
torch:2.1.0
torch:2.1.0
torchvision:0.1
4.1
torchvision:0.1
6.0
```
```
`Tips:以上dtk驱动、python、paddle等DCU相关工具版本需要严格一一对应`
`Tips:以上dtk驱动、python、paddle等DCU相关工具版本需要严格一一对应`
...
@@ -70,21 +71,13 @@ pip3 install -r requirements.txt
...
@@ -70,21 +71,13 @@ pip3 install -r requirements.txt
```
```
## 数据集
## 数据集
1.
通过
[
WIDER_FACE
](
http://113.200.138.88:18080/aidatasets/project-dependency/wider_face
)
下载所需数据集:
[
WIDER
_FACE
](
http://113.200.138.88:18080/aidatasets/project-dependency/wider_face
)
[
WIDER
Face Training Images
](
http://113.200.138.88:18080/aidatasets/project-dependency/wider_face
/-/blob/master/WIDER_train.zip
)
<div
align=
center
>
[
WIDER Face Validation Images
](
http://113.200.138.88:18080/aidatasets/project-dependency/wider_face/-/blob/master/WIDER_val.zip
)
<img
src=
"./doc/datasets.png"
/>
</div>
下载图片红框中三个数据并解压,也可直接点击下面链接直接下载:
[
WIDER Face Training Images(Tencent Drive)
](
https://share.weiyun.com/5WjCBWV
)
[
WIDER Face Validation Images(Tencent Drive)
](
https://share.weiyun.com/5ot9Qv1
)
[
WIDER Face Testing Images(Tencent Drive)
](
https://share.weiyun.com/5vSUomP
)
[
WIDER Face Testing Images
](
http://113.200.138.88:18080/aidatasets/project-dependency/wider_face/-/blob/master/WIDER_test.zip
)
数据集全部解压后的目录结构如下:
数据集全部解压后的目录结构如下:
...
@@ -108,7 +101,7 @@ pip3 install -r requirements.txt
...
@@ -108,7 +101,7 @@ pip3 install -r requirements.txt
```
```
解压完成后执行以下步骤:
解压完成后执行以下步骤:
1
.
将训练图片放置于 ./datasets/images/train的目录下,验证数据放置于./datasets/images/val目录下,存放目录结如下
2
.
将训练图片放置于 ./datasets/images/train的目录下,验证数据放置于./datasets/images/val目录下,存放目录结如下
```
```
├── images
├── images
│ ├── train
│ ├── train
...
@@ -121,8 +114,8 @@ pip3 install -r requirements.txt
...
@@ -121,8 +114,8 @@ pip3 install -r requirements.txt
│ └── 61--Street_Battle
│ └── 61--Street_Battle
```
```
2
.
如果是使用WIDER_train
、
WIDER_val数据, 可直接将./datasets/labels/下的train_wider_face.json重命名为train_face.json, val_wider_face.json重命名为val_face.json即可,无需进行标注文件格式转换;
3
.
如果是使用
`
WIDER_train
`
、
`
WIDER_val
`
数据, 可直接将
`
./datasets/labels/
`
下的
`
train_wider_face.json
`
重命名为
`
train_face.json
`
,
`
val_wider_face.json
`
重命名为
`
val_face.json
`
即可,无需进行标注文件格式转换;
反之,需要将训练图片/验证图片对应的人脸标注信息文件train.txt
/
val.txt,放置于
./datasets/annotations/下(train存放训练图片的标注文件,val存放验证图片的标注文件),存放目录结构如下:
反之,需要将训练图片/验证图片对应的人脸标注信息文件
`
train.txt
or
val.txt
`
,放置于
`
./datasets/annotations/
`
下(train存放训练图片的标注文件,val存放验证图片的标注文件),存放目录结构如下:
```
```
├── annotations
├── annotations
...
@@ -140,7 +133,7 @@ x, y, w, h, left_eye_x, left_eye_y, flag, right_eye_x, right_eye_y, flag, nose_x
...
@@ -140,7 +133,7 @@ x, y, w, h, left_eye_x, left_eye_y, flag, right_eye_x, right_eye_y, flag, nose_x
```
```
举个例子:
举个例子:
./datasets/annotations/train/train.txt
是
wider_face训练数据集的标注信息
`
./datasets/annotations/train/train.txt
`
是
`
wider_face
`
训练数据集的标注信息
```
```
# 0--Parade/0_Parade_marchingband_1_849.jpg
# 0--Parade/0_Parade_marchingband_1_849.jpg
...
@@ -148,7 +141,7 @@ x, y, w, h, left_eye_x, left_eye_y, flag, right_eye_x, right_eye_y, flag, nose_x
...
@@ -148,7 +141,7 @@ x, y, w, h, left_eye_x, left_eye_y, flag, right_eye_x, right_eye_y, flag, nose_x
...
...
```
```
3
.
生成训练所需的json格式标注数据:
4
.
生成训练所需的json格式标注数据:
```
bash
```
bash
cd
./datasets
cd
./datasets
python gen_data.py
python gen_data.py
...
@@ -158,7 +151,7 @@ python gen_data.py
...
@@ -158,7 +151,7 @@ python gen_data.py
## 训练
## 训练
默认训练模型保存在./exp/下,如需修改为自己的路径,可以对
`centerface_pytorch/src/lib/opts_pose.py`
的284行进行修改
默认训练模型保存在
`
./exp/
`
下,如需修改为自己的路径,可以对
`centerface_pytorch/src/lib/opts_pose.py`
的284行进行修改
。
### 单机单卡
### 单机单卡
```
bash
```
bash
...
@@ -203,7 +196,7 @@ WIDER_FACE验证集上的测试结果如下:
...
@@ -203,7 +196,7 @@ WIDER_FACE验证集上的测试结果如下:
教育,交通,公安,医疗
教育,交通,公安,医疗
## 源码仓库及问题反馈
## 源码仓库及问题反馈
https://developer.hpccube.com/codes/modelzoo/centerface_pytorch
-
https://developer.hpccube.com/codes/modelzoo/centerface_pytorch
## 参考资料
## 参考资料
https://github.com/chenjun2hao/CenterFace.pytorch
-
https://github.com/chenjun2hao/CenterFace.pytorch
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment