Vision Transformer ile Ateş Sınıflandırma

Bu projede Vision Transformer mimarisi kullanılmıştır. Vision Transformer, büyük boyutlu görüntüleri daha etkili bir şekilde temsil etmek için dikkat mekanizmalarını içerir.

Proje, ateş sınıflandırması üzerine olarak eğitilmiş bir Vision Transformer modelini içerir. Bu model, ateş veri kümesi üzerinde eğitilmiş ve ateşle ilişkilendirilmiş farklı durumları başarıyla tanıma yeteneğine sahiptir.

Projeyi yükleyin

    git clone https://github.com/VuralBayrakli/vision-transformer-classification.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 testing.py --model <model_path> --image <image_path>

Örnekler