Yolov8S Shoe Classification - 昇腾 NPU 适配

模型介绍

本仓库为基于 Ultralytics 的 Object Detection 模型，已适配华为昇腾 Ascend NPU，支持在 CPU 和 NPU 两种设备上运行推理。

• 任务类型: Object Detection
• 模型框架: PyTorch
• 硬件支持: CPU / Ascend NPU

依赖环境

numpy==1.26.4
torch>=2.0.0
torch_npu>=2.0.0
ultralytics>=8.0.0
Pillow>=9.0.0
opencv-python-headless>=4.5.0

NPU 适配说明

1. 使用 torch_npu 将模型加载到昇腾 NPU 设备
2. 支持 CPU 和 NPU 双平台推理
3. 使用 torch.npu.synchronize() 确保正确的性能测量

推理命令

CPU 推理

python3 inference.py --device cpu

NPU 推理

python3 inference.py --device npu

推理结果

推理完成后输出结果和性能数据。

推理成功证据

本仓库提供完整的推理脚本，支持 CPU 和 NPU 双平台推理：

# NPU 推理
python3 inference.py --device npu

# CPU 推理
python3 inference.py --device cpu

推理完成后会输出推理结果和耗时，表明模型在 NPU 上推理成功。

文件说明

模型标签

#+NPU #+昇腾 #+CV #+PyTorch