Anushdcosta commited on
Commit
31fcebe
·
verified ·
1 Parent(s): 4efc58e

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +32 -0
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