Vision Transformer ile Ateş Tespiti

Geleneksel nesne algılama yöntemlerinin ötesine geçen bir yaklaşım olan Vision Transformers yapısını içeren DINO modelini kullanarak gerçekleştirdiğim bu projede projede ateş tespitini sağlamaya çalıştım. Kendi veri setimi kullanarak DINO modelini ateş tespiti gerçekleştirecek şekilde eğittim.

Projeyi yükleyin

    git clone https://github.com/VuralBayrakli/vision-transformer-detection.git

DINO projesini indirin ve projenin kök dizine ekleyin

    git clone https://github.com/IDEA-Research/DINO.git

Modeli indirin

Eğitilmiş modeli buradan indirin ve projenin kök dizinine yerleştirin.

Gerekli kütüphaneleri yükleyin

    pip install -r requirements.txt

Modeli kullanın

    python inference.py --model <model_path> --image <image_path>

Test Görüntüleri