Segformer B0 Finetuned Flame

🚀 SegFormer (b0サイズ) モデル：FLAMEでファインチューニング済み

このモデルは、Forest Fire Detection というタイトルのディープラーニングプロジェクトで学習されました。

🚀 クイックスタート

このモデルは、画像セグメンテーションを通じた火災検出に使用されることを目的としています。

提供されている事前学習済みモデルは、FLAME データセットで学習率1e-3で3エポック間ファインチューニングされ、テスト例で 0.745 のIOUスコアを達成しました。

💻 使用例

基本的な使用法

このモデルを使って画像をセグメント化する方法は次の通りです。

from transformers import SegformerImageProcessor, SegformerForSemanticSegmentation
from PIL import Image
import requests

processor = AutoFeatureExtractor.from_pretrained("millionhz/segformer-b0-finetuned-flame")
model = SegformerForSemanticSegmentation.from_pretrained("millionhz/segformer-b0-finetuned-flame")


url = <add url here>
image = Image.open(requests.get(url, stream=True).raw)

inputs = feature_extractor(images=image, return_tensors="pt")
outputs = model(**inputs)

📄 ライセンス

このモデルのライセンスについては、こちら を参照してください。