Create README.md
Browse files
README.md
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
tags:
|
| 3 |
+
- image-classification
|
| 4 |
+
- transfer-learning
|
| 5 |
+
- keras
|
| 6 |
+
- tensorflow
|
| 7 |
+
- security
|
| 8 |
+
library_name: tensorflow
|
| 9 |
+
---
|
| 10 |
+
|
| 11 |
+
# Fine-Tuned Weapon vs. Not-Weapon Classifier (VGG16)
|
| 12 |
+
|
| 13 |
+
This repository contains a fine-tuned image classification model designed to distinguish between **"Weapon"** and **"Not-Weapon"** classes in images. The model was built using Transfer Learning on the VGG16 architecture.
|
| 14 |
+
|
| 15 |
+
## Model Structure and Training
|
| 16 |
+
|
| 17 |
+
* **Base Architecture:** VGG16 (pre-trained on ImageNet).
|
| 18 |
+
* **Training Method:** Transfer Learning (Feature Extraction) followed by **Fine-Tuning**.
|
| 19 |
+
* **File:** `weapon_classifier_final_tuned.keras` (This is the final, fine-tuned model file).
|
| 20 |
+
* **Input Size:** Images must be resized to **(224, 224)** pixels before prediction.
|
| 21 |
+
* **Output Activation:** Sigmoid (yielding a single probability value between 0 and 1).
|
| 22 |
+
|
| 23 |
+
## How to Use the Model (Inference)
|
| 24 |
+
|
| 25 |
+
To use this model, you need a Python environment with TensorFlow installed.
|
| 26 |
+
|
| 27 |
+
### 1. Installation
|
| 28 |
+
|
| 29 |
+
First, ensure you have the required libraries:
|
| 30 |
+
|
| 31 |
+
```bash
|
| 32 |
+
pip install tensorflow numpy Pillow
|