mixer_b16_224.miil_in21k 昇腾 NPU 适配

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch torch_npu timm Pillow numpy safetensors modelscope

python -c "from modelscope import snapshot_download; snapshot_download('timm/mixer_b16_224.miil_in21k')"

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

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

# CPU vs NPU精度对比
python compare_cpu_npu.py

import torch
import torch_npu
import timm
from PIL import Image
from timm.data import resolve_data_config, create_transform

# 加载模型到NPU
model = timm.create_model('mixer_b16_224.miil_in21k', pretrained=True).to('npu:0')
model.eval()

# 预处理
transform = create_transform(**resolve_data_config({}, model=model))
img = transform(Image.open('test.jpg').convert('RGB')).unsqueeze(0).to('npu:0')

# 推理
with torch.no_grad():
    torch.npu.synchronize()
    output = model(img)
    torch.npu.synchronize()
    probs = torch.nn.functional.softmax(output.cpu()[0], dim=0)

CPU Top-1: class 3963 (unknown)
NPU Top-1: class 3963 (unknown)
Top-1 Match: True
Top-5 Overlap: 5/5