许可证: 其他
标签:

• 训练生成
  基础模型: nvidia/mit-b0
  模型索引:
• 名称: 航拍无人机图像分割
  结果: []
  任务标签: 图像分割

航拍无人机图像分割模型

本模型是基于nvidia/mit-b0微调的版本，在评估集上取得以下结果：

• 损失值: 0.8852
• 平均交并比(mIoU): 0.2994
• 平均准确率: 0.3923
• 整体准确率: 0.7774

模型描述

（待补充详细信息）

训练超参数

训练过程中使用的超参数配置：

• 学习率: 5e-05
• 训练批次大小: 24
• 评估批次大小: 8
• 随机种子: 42
• 优化器: Adam（β1=0.9，β2=0.999，ε=1e-08）
• 学习率调度器类型: 线性
• 训练轮次: 50

评估结果

{
    'mean_iou': 0.27989828118195953,
    'mean_accuracy': 0.3712316062110093,
    'overall_accuracy': 0.7671712239583334,
    'per_category_iou': [nan, 0.856, 0.322, 0.769, 0.575, 0.439, 0.001, 0.141, 0.518, 0.770, 0.274, 0.0, 0.0, 0.0, 0.0, 0.058, 0.0, 0.456, 0.0, 0.259, 0.488, 0.0, 0.233, nan],
    'per_category_accuracy': [nan, 0.962, 0.440, 0.870, 0.815, 0.497, 0.001, 0.144, 0.810, 0.887, 0.468, 0.0, 0.0, 0.0, 0.0, 0.059, 0.0, 0.556, 0.0, 0.384, 0.761, 0.0, 0.513, nan]
}

训练过程可视化

框架版本

• Transformers 4.38.1
• PyTorch 2.1.2
• Datasets 2.1.0
• Tokenizers 0.15.2