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