🚀 segformer-b3-ade-512-512-finetuned-coastTrain
该模型是 nvidia/segformer-b3-finetuned-ade-512-512 在 peldrak/coastTrain_512-512 数据集上的微调版本,可用于图像分割任务。
🚀 快速开始
此模型是 nvidia/segformer-b3-finetuned-ade-512-512 在 peldrak/coastTrain_512-512 数据集上的微调版本。它在评估集上取得了以下结果:
- 损失率 (Loss):0.7613
- 平均交并比 (Mean Iou):0.7092
- 平均准确率 (Mean Accuracy):0.8104
- 总体准确率 (Overall Accuracy):0.8790
- 水体准确率 (Accuracy Water):0.9352
- 白浪准确率 (Accuracy Whitewater):0.8067
- 沉积物准确率 (Accuracy Sediment):0.8732
- 其他自然地形准确率 (Accuracy Other Natural Terrain):0.5054
- 植被准确率 (Accuracy Vegetation):0.8997
- 建筑准确率 (Accuracy Development):0.8714
- 未知类别准确率 (Accuracy Unknown):0.7814
- 水体交并比 (Iou Water):0.8677
- 白浪交并比 (Iou Whitewater):0.6795
- 沉积物交并比 (Iou Sediment):0.7649
- 其他自然地形交并比 (Iou Other Natural Terrain):0.4259
- 植被交并比 (Iou Vegetation):0.7883
- 建筑交并比 (Iou Development):0.7211
- 未知类别交并比 (Iou Unknown):0.7170
🔧 技术细节
训练超参数
在训练过程中使用了以下超参数:
- 学习率 (learning_rate):6e-05
- 训练批次大小 (train_batch_size):4
- 评估批次大小 (eval_batch_size):4
- 随机种子 (seed):42
- 优化器 (optimizer):Adam,β1=0.9,β2=0.999,ε=1e-08
- 学习率调度器类型 (lr_scheduler_type):线性
- 训练轮数 (num_epochs):25
训练结果
以下是训练过程中的详细结果:
训练损失 |
轮数 |
步数 |
验证损失 |
平均交并比 |
平均准确率 |
总体准确率 |
水体准确率 |
白浪准确率 |
沉积物准确率 |
其他自然地形准确率 |
植被准确率 |
建筑准确率 |
未知类别准确率 |
水体交并比 |
白浪交并比 |
沉积物交并比 |
其他自然地形交并比 |
植被交并比 |
建筑交并比 |
未知类别交并比 |
1.7642 |
0.05 |
20 |
1.6699 |
0.1741 |
0.2887 |
0.4511 |
0.3629 |
0.3020 |
0.0122 |
0.0013 |
0.8998 |
0.1317 |
0.3106 |
0.3310 |
0.0708 |
0.0112 |
0.0013 |
0.3953 |
0.1007 |
0.3084 |
1.6158 |
0.11 |
40 |
1.3903 |
0.1804 |
0.2783 |
0.5516 |
0.6957 |
0.0198 |
0.1032 |
0.0000 |
0.9605 |
0.1077 |
0.0616 |
0.5309 |
0.0184 |
0.0965 |
0.0000 |
0.4589 |
0.0973 |
0.0606 |
1.3168 |
0.16 |
60 |
1.1710 |
0.2583 |
0.3483 |
0.6425 |
0.8324 |
0.0359 |
0.0669 |
0.0 |
0.9578 |
0.1157 |
0.4296 |
0.6688 |
0.0344 |
0.0630 |
0.0 |
0.5089 |
0.1094 |
0.4233 |
... |
... |
... |
... |
... |
... |
... |
... |
... |
... |
... |
... |
... |
... |
... |
... |
... |
... |
... |
... |
... |
📄 许可证
该模型使用的许可证为 other。
🔍 框架版本
- Transformers:4.37.0
- Pytorch:2.1.2
- Datasets:2.17.1
- Tokenizers:0.15.1