Ascend-SACT/YOLO-World
模型介绍文件和版本Pull Requests讨论分析
下载使用量0

前言

镜像内已预置了适配昇腾平台的YOLO-World,仅需启动测试脚本,即可启动推理过程。具体操作步骤如下。

使用前提

  • 宿主机上已经安装好固件与驱动,具体可参考安装驱动和固件。
  • 宿主机上已经安装好Docker。

使用方法

  1. 下载镜像 请先下载本项目内的镜像,其中包含了 YOLO-World 代码和 yolov8x-worldv2.pt 权重。用户也可以自行替换权重。
  2. 启动容器
IMAGE_ID="your image ID"
container_name="your container name"
docker run \
--net=host \
--shm-size=512g \
--name $container_name \
--privileged \
-v /usr/local/dcmi:/usr/local/dcmi \
-v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
-v /usr/local/Ascend/driver:/usr/local/Ascend/driver:ro \
-v /usr/local/sbin:/usr/local/sbin:ro \
-v /etc/ascend_install.info:/etc/ascend_install.info \
-it $IMAGE_ID bash
  1. 路径说明 YOLO-World的推理代码和权重已经内置在容器内 1)代码路径 /home/yolo_check 2)权重路径  /home/yolov8x-worldv2.pt  
  2. 启动推理
cd /home/yolo_check
python yolo_world_xn.py