Subh775

The paradigm has shifted! While CNNs traditionally dominated object detection with faster inference times, RF-DETR (Roboflow's Detection Transformer) has revolutionized the field. This transformer-based architecture not only outperforms CNNs in accuracy but also delivers faster inference for real-time applications. This repository contains a fine-tuned RF-DETR Nano model specifically trained for threat detection, capable of identifying four critical threat categories with high precision and speed. RF-DETR Threat Detection is a specialized computer vision model designed for security and surveillance applications. Built on Roboflow's cutting-edge RF-DETR architecture, this model can accurately detect and classify potential threats in real-time scenarios. | Class ID | Threat Type | Description | |----------|-------------|-------------| | 1 | Gun | Any type of firearm weapon including pistols, rifles, and other firearms | | 2 | Explosive | Fire, explosion scenarios, and explosive devices | | 3 | Grenade | Hand grenades and similar explosive devices | | 4 | Knife | Bladed weapons including knives, daggers, and sharp objects | Our custom threat detection dataset was meticulously curated and annotated to ensure robust model performance across diverse scenarios. The model is trained to detect threats across various scales, from small concealed weapons to larger explosive devices. The training process demonstrates excellent convergence with: - Consistent loss reduction over 50 epochs - Stable validation performance indicating good generalization - Balanced precision and recall across all threat categories | Metric | Gun | Explosive | Grenade | Knife | Overall | |--------|-----|-----------|---------|-------|-------------| | mAP@50:95 | 62.3% | 47.2% | 80.5% | 54.4% | 61.1% | | mAP@50 | 90.1% | 69.6% | 93.7% | 85.8% | 84.8% | | Precision | 92.4% | 54.6% | 97.2% | 91.1% | 83.8% | | Recall | 85.0% | 85.0% | 85.0% | 85.0% | 85.0% | | Metric | Gun | Explosive | Grenade | Knife | Overall | |--------|-----|-----------|---------|-------|-------------| | mAP@50:95 | 65.3% | 35.7% | 83.2% | 49.8% | 58.5% | | mAP@50 | 93.1% | 60.5% | 91.1% | 79.7% | 81.1% | | Precision | 96.7% | 49.7% | 93.1% | 86.5% | 81.5% | | Recall | 83.0% | 83.0% | 83.0% | 83.0% | 83.0% | - 84.8% mAP@50 on validation set - Fast inference with RF-DETR Nano architecture - Excellent precision for Gun (96.7%) and Grenade (93.1%) detection - Consistent recall of 83-85% across all threat categories - Robust generalization from validation to test performance - Base Architecture: RF-DETR Nano - Input Resolution: 640×640 pixels - Backbone: Optimized transformer encoder - Detection Head: Custom 4-class threat detection - Inference Speed: ~50ms per image (GPU) - Model Size: Lightweight for edge deployment Training Configuration - Epochs: 50 - Batch Size: Optimized for available GPU memory - Optimizer: AdamW with learning rate scheduling - Data Augmentation: Advanced augmentation pipeline for robust training - Loss Function: Multi-scale detection loss with class balancing Training Strategy 1. Progressive Training: Started with lower resolution, gradually increased 2. Class Balancing: Weighted loss to handle class imbalance 3. Data Augmentation: Extensive augmentation to improve generalization 4. Early Stopping: Monitored validation mAP to prevent overfitting - You can use: videoprocessing.py to process large videos - Roboflow for the RF-DETR architecture - Hugging Face for model hosting and distribution - PyTorch ecosystem for deep learning framework - Supervision library for computer vision utilities Disclaimer: This model is designed for research purpose only. It's predictions cannot be taken into account for deployment right now.