Aerial Drone Image Segmentation

🚀 无人机航拍图像分割模型

该模型用于无人机航拍图像分割，是基于 nvidia/mit-b0 微调得到的模型。在评估集上取得了不错的效果，能有效对航拍图像进行分割处理，为相关领域的图像分析提供支持。

📚 详细文档

模型描述

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

• 损失值：0.8852
• 平均交并比（Mean Iou）：0.2994
• 平均准确率（Mean Accuracy）：0.3923
• 整体准确率（Overall Accuracy）：0.7774

训练超参数

训练过程中使用了以下超参数：

评估结果

{'mean_iou': 0.27989828118195953,
 'mean_accuracy': 0.3712316062110093,
 'overall_accuracy': 0.7671712239583334,
 'per_category_iou': array([       nan, 0.8560476 , 0.32234631, 0.76880948, 0.57517691,
        0.43877125, 0.00114888, 0.14091442, 0.51807365, 0.76964765,
        0.27391949, 0.        , 0.        , 0.        , 0.        ,
        0.05778175, 0.        , 0.45566807, 0.        , 0.25864545,
        0.48767764, 0.        , 0.23313364,        nan]),
 'per_category_accuracy': array([       nan, 0.96170675, 0.43993514, 0.86977593, 0.8149788 ,
        0.49739671, 0.00114987, 0.14445379, 0.80978302, 0.88661108,
        0.46787116, 0.        , 0.        , 0.        , 0.        ,
        0.05947339, 0.        , 0.55639324, 0.        , 0.38358184,
        0.761303  , 0.        , 0.51268161,        nan])}

训练结果

框架版本

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

📄 许可证

该模型使用其他许可证。