AIbase
EN
Home
Vit Fire Detection
V

Vit Fire Detection

Developed by EdBianchi
A vision Transformer model fine-tuned from google/vit-base-patch16-224-in21k, specialized in fire detection tasks, demonstrating high precision and recall rates on the evaluation dataset.
Image Classification
Transformers
Open Source License:Apache-2.0#Fire Detection#High-Precision Vision#ViT Fine-Tuning
Downloads 1,851
Release Time : 12/29/2022

Model Overview

This model is a fire detection model based on the Vision Transformer architecture, implemented by fine-tuning the google/vit-base-patch16-224-in21k pre-trained model, suitable for flame recognition tasks in images.

Model Features

High-Precision Detection
Achieves 99.6% precision and recall rates on the evaluation dataset, demonstrating excellent flame recognition capabilities.
ViT-Based Architecture
Utilizes the Vision Transformer architecture, leveraging self-attention mechanisms to capture global image features.
Efficient Fine-Tuning
Fine-tuned from a pre-trained model, requiring only a small amount of training data to achieve good performance.

Model Capabilities

Image Classification
Fire Detection
Visual Feature Extraction

Use Cases

Security Monitoring
Fire Warning System
Used for real-time flame detection in surveillance camera footage
Can quickly identify fire hazards with 99.6% accuracy
Industrial Safety
Workshop Fire Monitoring
Detects abnormal flames in industrial production environments
Featured Recommended AI Models
AIbase
Empowering the Future, Your AI Solution Knowledge Base
EnglishįŽ€äŊ“中文įšéĢ”中文ãĢãģんご
Š 2025AIbase