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
5ee8260c
Commit
5ee8260c
authored
Nov 08, 2023
by
chenych
Browse files
modify README
parent
49a371b2
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
22 additions
and
4 deletions
+22
-4
README.md
README.md
+22
-3
docker/Dockerfile
docker/Dockerfile
+0
-1
image.png
image.png
+0
-0
process.png
process.png
+0
-0
No files found.
README.md
View file @
5ee8260c
...
@@ -11,27 +11,46 @@ CenterFace是一种人脸检测算法,采用了轻量级网络mobileNetV2作
...
@@ -11,27 +11,46 @@ CenterFace是一种人脸检测算法,采用了轻量级网络mobileNetV2作
## 算法原理
## 算法原理
CenterFace模型是一种基于单阶段人脸检测算法,作者借鉴了CenterNet的思想,将人脸检测转换为标准点问题,根据人脸中心点来回归人脸框的大小和五个标志点。
CenterFace模型是一种基于单阶段人脸检测算法,作者借鉴了CenterNet的思想,将人脸检测转换为标准点问题,根据人脸中心点来回归人脸框的大小和五个标志点。
| 符号 | 说明 |
|:--------:| :--------:|
| F | Feature Map |
| c | Confidence |
| x_c,y_c | 人脸框中心点的x,y坐标 |
| x_l,y_l | 人脸框左上角点的x,y坐标 |
| x_r,y_r | 人脸框右下角点的x,y坐标 |
<div
align=
center
>
<img
src=
"./process.png"
/>
</div>
## 环境配置
## 环境配置
### Docker(方法一)
### Docker(方法一)
-v 路径、docker_name和imageID根据实际情况修改
```
```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04
.1
-py38-latest
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py38-latest
docker run docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
docker run docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
cd /
path/workspace
/
cd /
your_code_path/centerface_pytorch
/
pip3 install -r requirements.txt
pip3 install -r requirements.txt
```
```
### Dockerfile(方法二)
### Dockerfile(方法二)
-v 路径、docker_name和imageID根据实际情况修改
```
```
cd ./docker
cd ./docker
cp ../requirements.txt requirements.txt
docker build --no-cache -t centerface:latest .
docker build --no-cache -t centerface:latest .
docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
```
```
### Anaconda(方法三)
### Anaconda(方法三)
...
@@ -39,7 +58,7 @@ docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --p
...
@@ -39,7 +58,7 @@ docker run -it -v /path/your_code_data/:/path/your_code_data/ --shm-size=32G --p
1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装: https://developer.hpccube.com/tool/
1、关于本项目DCU显卡所需的特殊深度学习库可从光合开发者社区下载安装: https://developer.hpccube.com/tool/
```
```
DTK软件栈:dtk23.04
.1
DTK软件栈:dtk23.04
python:python3.8
python:python3.8
torch:1.13.1
torch:1.13.1
torchvision:0.14.1
torchvision:0.14.1
...
...
docker/Dockerfile
View file @
5ee8260c
...
@@ -4,4 +4,3 @@ COPY requirments.txt requirments.txt
...
@@ -4,4 +4,3 @@ COPY requirments.txt requirments.txt
RUN
pip3
install
-r
requirements.txt
RUN
pip3
install
-r
requirements.txt
ADD
https://ultralytics.com/assets/Arial.ttf /root/.config/Ultralytics/
ADD
https://ultralytics.com/assets/Arial.ttf /root/.config/Ultralytics/
image.png
deleted
100644 → 0
View file @
49a371b2
23.7 KB
process.png
0 → 100644
View file @
5ee8260c
8.52 KB
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