Skip to content

GitLab

Commit 58b5daa5 authored by liucong's avatar liucong
Browse files

更新示例工程

parent 36e49304
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 1 deletion
README.md
View file @ 58b5daa5
...@@ -20,6 +20,8 @@ RetinaFace模型有几个主要特点： ...@@ -20,6 +20,8 @@ RetinaFace模型有几个主要特点：
- 引入 SSH 算法的 Context Modeling； - 引入 SSH 算法的 Context Modeling；
- 多任务训练，提供额外的监督信息。 - 多任务训练，提供额外的监督信息。
<img src="./Doc/Image/RetinaFace_04.png" style="zoom:70%;" align=center>
## 环境配置 ## 环境配置
### Docker（方法一） ### Docker（方法一）
...@@ -46,6 +48,9 @@ cd ./docker ...@@ -46,6 +48,9 @@ cd ./docker
docker build --no-cache -t retinaface_migraphx:2.0 . 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 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
# 激活dtk
source /opt/dtk/env.sh
``` ```
## 数据集 ## 数据集
......
docker/Dockerfile
View file @ 58b5daa5
FROM image.sourcefind.cn:5000/dcu/admin/base/migraphx:4.0.0-centos7.6-dtk23.04.1-py38-latest 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
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