Aerial Drone Image Segmentation

🚀 空中ドローン画像セグメンテーション

このモデルは、画像セグメンテーションタスクに特化したモデルです。nvidia/mit - b0 をファインチューニングしたもので、評価セットで良好な結果を達成しています。

📚 詳細ドキュメント

モデルの説明

このモデルは nvidia/mit-b0 のファインチューニング版です。評価セットでは以下の結果を達成しています。

• 損失: 0.8852
• 平均IoU: 0.2994
• 平均精度: 0.3923
• 全体精度: 0.7774

トレーニングハイパーパラメータ

トレーニング時には以下のハイパーパラメータが使用されました。

• 学習率: 5e - 05
• トレーニングバッチサイズ: 24
• 評価バッチサイズ: 8
• シード: 42
• オプティマイザ: Adam (betas=(0.9,0.999), epsilon=1e - 08)
• 学習率スケジューラの種類: linear
• エポック数: 50

評価結果

{'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

📄 ライセンス

Other