diff --git a/README.md b/README.md index 48919a9..6a27bd9 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,4 @@ -# enginex-hygon-tts +# enginex-hygon-OD 运行于【海光 DCU】系列算力卡的【目标检测】引擎 @@ -14,4 +14,7 @@ - AI-ModelScope/Yolov10 - merve/yolov9 - AI-ModelScope/YOLO11 -- AI-ModelScope/GroundingDINO \ No newline at end of file +- AI-ModelScope/GroundingDINO +- yolo_master/yolov3 +- atalaydenknalbant/Yolov13 +- amd/retinaface \ No newline at end of file diff --git a/enginex/Yolov13.md b/enginex/Yolov13.md new file mode 100644 index 0000000..b464510 --- /dev/null +++ b/enginex/Yolov13.md @@ -0,0 +1,10 @@ +# 运行方式 + +```python +docker pull git.modelhub.org.cn:9443/enginex-hygon/pytorch:2.4.1-ubuntu22.04-dtk25.04.1-py3.10 +# 为以上拉取的docker的镜像ID替换,本镜像为:e50d644287fd +docker run -it --shm-size=64G -v $PWD/yolov13:/home/yolov13 -v /opt/hyhal:/opt/hyhal:ro --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name yolov13 bash +cd /home/yolov13 +pip install -r requirements.txt # requirements.txt +pip install -e . # ultralytics==8.3.63 +``` diff --git a/enginex/retinaface.md b/enginex/retinaface.md index f35328b..1eb25ed 100644 --- a/enginex/retinaface.md +++ b/enginex/retinaface.md @@ -1,5 +1,10 @@ # 运行方式 ```python -docker pull git.modelhub.org.cn:9443/enginex-hygon/ +docker pull git.modelhub.org.cn:9443/enginex-hygon/migraphx:4.3.0-ubuntu20.04-dtk24.04.1-py3.8 +docker run --shm-size 16g --network=host --name=retinaface_migraphx -v /opt/hyhal:/opt/hyhal:ro --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 /bin/bash + +# 激活dtk +source /opt/dtk/env.sh + ``` \ No newline at end of file diff --git a/enginex/yolov3.md b/enginex/yolov3.md new file mode 100644 index 0000000..51b6f22 --- /dev/null +++ b/enginex/yolov3.md @@ -0,0 +1,9 @@ +# 运行方式 + +```python +docker pull git.modelhub.org.cn:9443/enginex-hygon/migraphx:4.3.0-ubuntu20.04-dtk24.04.1-py3.10 +docker run --shm-size 16g --network=host --name=yolov3_migraphx --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v $PWD/yolov3_migraphx:/home/yolov3_migraphx -v /opt/hyhal:/opt/hyhal:ro -it /bin/bash + +# 激活dtk +source /opt/dtk/env.sh +``` \ No newline at end of file