当前位置：首页 > news >正文

如何扩展识别类别？基于阿里万物识别模型的微调方法探索

news 2026/8/23 10:18:34

如何扩展识别类别？基于阿里万物识别模型的微调方法探索

核心价值：本文将带你从零开始，基于阿里开源的“万物识别-中文-通用领域”模型，实现自定义类别的扩展与微调。不仅涵盖推理部署流程，更深入讲解如何构建数据集、修改分类头、训练适配新场景的识别模型，并提供完整可运行代码与工程优化建议。

一、背景与问题提出：为什么需要扩展识别类别？

在智能视觉应用中，通用图像分类模型（如ImageNet预训练模型）虽能识别上千类常见物体，但在特定行业或垂直场景下往往力不从心。例如：

工业质检需识别“螺母松动”“焊点虚焊”等细粒度缺陷
零售货架监控要区分“某品牌矿泉水500ml vs 380ml”
农业监测需判断“玉米叶斑病”“小麦锈病”等专业病害

这些需求超出了通用模型的原始设计范畴。而阿里云推出的“万物识别-中文-通用领域”模型，作为一款面向中文语境、支持细粒度语义理解的开源图像识别系统，为定制化扩展提供了良好基础。

但该模型默认输出的是其训练时定义的固定类别集合。若想让它识别我们新增的“白令海产包装盒”“AI实验记录本”等个性化对象，就必须进行模型微调（Fine-tuning）。

本文目标：
✅ 掌握阿里万物识别模型的本地部署与推理流程
✅ 构建适用于微调的数据集格式
✅ 修改模型结构以适配新类别数量
✅ 实现端到端的微调训练与评估
✅ 提供可复用的最佳实践建议

二、环境准备与推理验证：跑通 baseline 流程

在动手微调前，先确保基础环境可用，并成功运行原模型推理。

环境依赖说明

根据项目要求，已配置如下环境：

# 基础环境信息 Python: 3.11 (conda env: py311wwts) PyTorch: 2.5 CUDA: 12.1 (如有GPU)

依赖文件位于/root/requirements.txt，可通过以下命令安装（如需重建环境）：

pip install -r /root/requirements.txt

激活环境并复制工作文件

# 激活指定 conda 环境 conda activate py311wwts # 复制推理脚本和示例图片至工作区 cp /root/推理.py /root/workspace/ cp /root/bailing.png /root/workspace/ # 进入工作目录 cd /root/workspace

⚠️ 注意：复制后需手动修改推理.py中的图像路径为/root/workspace/bailing.png

执行推理测试

运行原始推理脚本：

python 推理.py

预期输出应包含类似结果：

预测类别: 白令海产 置信度: 0.987

这表明模型已正确加载并完成前向推理，具备进一步微调的基础条件。

三、数据准备：构建符合微调需求的图像数据集

微调的第一步是准备高质量、标注清晰的训练数据。

数据组织结构规范

遵循 PyTorch 标准ImageFolder结构：

dataset/ ├── new_class_1/ │ ├── img1.jpg │ └── img2.jpg ├── new_class_2/ │ ├── photo1.png │ └── photo2.png └── ...

例如我们要增加两个新类别：

dataset/ ├── 白令海产礼盒/ │ ├── bailing_gift_01.jpg │ └── bailing_gift_02.jpg ├── AI实验手册/ │ ├── labnote_01.jpg │ └── labnote_02.jpg

每类建议至少 50 张样本，保证多样性（角度、光照、背景变化）。

数据预处理策略

使用torchvision.transforms进行标准化增强：

from torchvision import transforms train_transform = transforms.Compose([ transforms.Resize((224, 224)), transforms.RandomHorizontalFlip(p=0.5), transforms.ColorJitter(brightness=0.2, contrast=0.2), transforms.ToTensor(), transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) ]) val_transform = transforms.Compose([ transforms.Resize((224, 224)), transforms.ToTensor(), transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) ])

四、模型结构解析与分类头替换

“万物识别-中文-通用领域”模型本质上是一个基于 Vision Transformer 或 ResNet 主干的分类网络。我们需要解剖其结构，定位可修改部分。

查看模型结构（假设通过 Hugging Face 或本地加载）

import torch from models import get_model # 假设模型入口函数 model = get_model(pretrained=True) print(model)

观察输出中的分类层，通常命名为：

classifier
head
fc
cls_head

假设发现最后一层为：

(model.head): Linear(in_features=768, out_features=1000, bias=True)

其中out_features=1000表示原模型支持 1000 类。现在我们要将其改为支持 1002 类（原有 + 新增 2 类）。

替换分类头代码实现

# 冻结主干网络参数（可选） for param in model.parameters(): param.requires_grad = False # 仅替换最后的全连接层 num_new_classes = 2 original_in_features = model.head.in_features model.head = torch.nn.Linear(original_in_features, num_new_classes) # 将模型移动到 GPU（如果可用） device = torch.device("cuda" if torch.cuda.is_available() else "cpu") model.to(device)

✅最佳实践建议：初期只微调分类头；待收敛后再解冻部分主干层进行联合训练。

五、微调训练流程：完整代码实现

以下是完整的微调训练脚本，保存为finetune.py。

# finetune.py import torch import torch.nn as nn from torch.utils.data import DataLoader from torchvision import datasets, transforms from models import get_model import os # ---------------------------- # 1. 超参数设置 # ---------------------------- data_dir = '/root/workspace/dataset' batch_size = 16 num_epochs = 10 learning_rate = 1e-3 save_path = '/root/workspace/checkpoint.pth' # ---------------------------- # 2. 数据加载 # ---------------------------- transform = transforms.Compose([ transforms.Resize((224, 224)), transforms.ToTensor(), transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]) ]) dataset = datasets.ImageFolder(data_dir, transform=transform) dataloader = DataLoader(dataset, batch_size=batch_size, shuffle=True, num_workers=4) print(f"共加载 {len(dataset)} 张图像，{len(dataset.classes)} 个类别：{dataset.classes}") # ---------------------------- # 3. 模型构建 # ---------------------------- model = get_model(pretrained=True) # 替换分类头 num_new_classes = len(dataset.classes) in_features = model.head.in_features model.head = nn.Linear(in_features, num_new_classes) model.to(device) # ---------------------------- # 4. 损失函数与优化器 # ---------------------------- criterion = nn.CrossEntropyLoss() optimizer = torch.optim.Adam(model.head.parameters(), lr=learning_rate) # 仅训练新头 # ---------------------------- # 5. 训练循环 # ---------------------------- device = torch.device("cuda" if torch.cuda.is_available() else "cpu") model.train() for epoch in range(num_epochs): running_loss = 0.0 correct = 0 total = 0 for images, labels in dataloader: images, labels = images.to(device), labels.to(device) optimizer.zero_grad() outputs = model(images) loss = criterion(outputs, labels) loss.backward() optimizer.step() running_loss += loss.item() _, predicted = outputs.max(1) total += labels.size(0) correct += predicted.eq(labels).sum().item() acc = 100. * correct / total print(f"Epoch [{epoch+1}/{num_epochs}], Loss: {running_loss:.4f}, Acc: {acc:.2f}%") # ---------------------------- # 6. 保存模型 # ---------------------------- torch.save(model.state_dict(), save_path) print(f"模型已保存至 {save_path}")

六、推理脚本改造：加载微调后模型进行预测

创建新的推理脚本custom_inference.py，用于加载微调后的模型。

# custom_inference.py from PIL import Image import torch from torchvision import transforms from models import get_model # 加载模型 model = get_model(pretrained=False) # 不加载预训练权重 num_new_classes = 2 model.head = torch.nn.Linear(model.head.in_features, num_new_classes) model.load_state_dict(torch.load('/root/workspace/checkpoint.pth')) model.eval() # 类别映射 class_names = ['白令海产礼盒', 'AI实验手册'] # 图像预处理 transform = transforms.Compose([ transforms.Resize((224, 224)), transforms.ToTensor(), transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]) ]) # 推理 image_path = '/root/workspace/test.jpg' image = Image.open(image_path).convert('RGB') input_tensor = transform(image).unsqueeze(0) with torch.no_grad(): output = model(input_tensor) probabilities = torch.nn.functional.softmax(output[0], dim=0) conf, pred = torch.max(probabilities, 0) print(f"预测类别: {class_names[pred.item()]}") print(f"置信度: {conf.item():.3f}")

七、关键挑战与优化建议

❗ 常见问题及解决方案

| 问题 | 原因分析 | 解决方案 | |------|---------|----------| | 微调后准确率低 | 新数据量少，过拟合风险高 | 使用更强的数据增强、添加 Dropout、早停机制 | | 模型无法收敛 | 学习率过高或优化器选择不当 | 降低学习率至 1e-4 ~ 1e-5，改用 SGD with momentum | | 显存不足 | Batch size 过大或模型太大 | 减小 batch_size，启用梯度累积（gradient accumulation） |

🚀 性能优化建议

渐进式解冻（Progressive Unfreezing）
先训练分类头 → 再解冻最后几层 Transformer Block → 最后微调整个网络
使用混合精度训练
减少显存占用，提升训练速度：

python scaler = torch.cuda.amp.GradScaler() with torch.cuda.amp.autocast(): outputs = model(images) loss = criterion(outputs, labels) scaler.scale(loss).backward() scaler.step(optimizer) scaler.update()