Commit 36e49304 authored by liucong's avatar liucong
Browse files

修改readme

parent 195c1008
...@@ -22,7 +22,7 @@ RetinaFace模型有几个主要特点: ...@@ -22,7 +22,7 @@ RetinaFace模型有几个主要特点:
## 环境配置 ## 环境配置
### Docker ### Docker(方法一)
拉取镜像: 拉取镜像:
...@@ -39,6 +39,15 @@ docker run --shm-size 16g --network=host --name=retinaface_migraphx --privileged ...@@ -39,6 +39,15 @@ docker run --shm-size 16g --network=host --name=retinaface_migraphx --privileged
source /opt/dtk/env.sh source /opt/dtk/env.sh
``` ```
### Dockerfile(方法二)
```
cd ./docker
docker build --no-cache -t retinaface_migraphx:2.0 .
docker run --shm-size 16g --network=host --name=retinaface_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/retinaface_migraphx:/home/retinaface_migraphx -it <Your Image ID> /bin/bash
```
## 数据集 ## 数据集
根据提供的样本数据,进行人脸检测。 根据提供的样本数据,进行人脸检测。
...@@ -57,8 +66,6 @@ export PYTHONPATH=/opt/dtk/lib:$PYTHONPATH ...@@ -57,8 +66,6 @@ export PYTHONPATH=/opt/dtk/lib:$PYTHONPATH
#### 安装依赖 #### 安装依赖
- 安装DTK版的Pytorch和torchvision,下载地址:https://cancon.hpccube.com:65024/4/main/pytorch,https://cancon.hpccube.com:65024/4/main/vision
``` ```
# 进入python示例目录 # 进入python示例目录
cd <path_to_retinaface_migraphx>/Python cd <path_to_retinaface_migraphx>/Python
...@@ -129,6 +136,10 @@ C++程序运行结束后,会在build目录生成RetinaFace人脸检测结果 ...@@ -129,6 +136,10 @@ C++程序运行结束后,会在build目录生成RetinaFace人脸检测结果
<img src="./Resource/Images/Result_1.jpg" alt="Result" style="zoom:67%;" /> <img src="./Resource/Images/Result_1.jpg" alt="Result" style="zoom:67%;" />
### 精度
## 应用场景 ## 应用场景
### 算法类别 ### 算法类别
...@@ -143,6 +154,6 @@ C++程序运行结束后,会在build目录生成RetinaFace人脸检测结果 ...@@ -143,6 +154,6 @@ C++程序运行结束后,会在build目录生成RetinaFace人脸检测结果
https://developer.hpccube.com/codes/modelzoo/retinaface_migraphx https://developer.hpccube.com/codes/modelzoo/retinaface_migraphx
## 参考 ## 参考资料
https://github.com/biubug6/Pytorch_Retinaface https://github.com/biubug6/Pytorch_Retinaface
FROM image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.0.0-centos7.6-dtk23.04.1-py38-latest
RUN source /opt/dtk/env.sh
...@@ -5,6 +5,6 @@ modelName=retinaface_migraphx ...@@ -5,6 +5,6 @@ modelName=retinaface_migraphx
#模型描述 #模型描述
modelDescription=RetinaFace是一个经典的人脸检测模型,采用了SSD架构。 modelDescription=RetinaFace是一个经典的人脸检测模型,采用了SSD架构。
#应用场景 #应用场景
appScenario=推理,CV,人脸检测 appScenario=推理,人脸检测,金融,交通,教育
#框架类型 #框架类型
frameType=migraphx frameType=migraphx
\ No newline at end of file
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment