Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
# 🚗🏍️ Car vs Bike Image Classifier
|
| 2 |
|
| 3 |
This repository contains a Convolutional Neural Network (CNN) model trained to classify whether an image contains a **car** or a **bike**.
|
|
@@ -5,9 +27,7 @@ This repository contains a Convolutional Neural Network (CNN) model trained to c
|
|
| 5 |
## 📊 Model Overview
|
| 6 |
|
| 7 |
- **Input:** RGB image resized to 128x128
|
| 8 |
-
- **Output:** Binary classification
|
| 9 |
-
- `1` → Car
|
| 10 |
-
- `0` → Bike
|
| 11 |
- **Framework:** TensorFlow / Keras
|
| 12 |
- **Model Format:** `.h5` (HDF5)
|
| 13 |
|
|
@@ -45,30 +65,35 @@ if prediction > 0.5:
|
|
| 45 |
print("Prediction: 🚗 Car")
|
| 46 |
else:
|
| 47 |
print("Prediction: 🏍️ Bike")
|
|
|
|
| 48 |
|
|
|
|
| 49 |
|
| 50 |
-
|
| 51 |
-
|
|
|
|
| 52 |
|
| 53 |
-
|
| 54 |
|
|
|
|
| 55 |
|
| 56 |
-
|
| 57 |
-
You can clone and use this model like so:
|
| 58 |
-
|
| 59 |
git lfs install
|
| 60 |
-
git clone https://huggingface.co/
|
|
|
|
|
|
|
|
|
|
| 61 |
|
|
|
|
|
|
|
|
|
|
| 62 |
|
| 63 |
-
|
| 64 |
-
Python 3.7+
|
| 65 |
-
TensorFlow ≥ 2.5
|
| 66 |
-
NumPy
|
| 67 |
-
Keras
|
| 68 |
|
|
|
|
|
|
|
|
|
|
| 69 |
|
| 70 |
-
|
| 71 |
-
Neeraja Dakkata
|
| 72 |
-
📧 dakkataneeraja0031@gmail.com
|
| 73 |
-
🔗 GitHub: https://github.com/D-Neeraja
|
| 74 |
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
---
|
| 3 |
+
license: mit
|
| 4 |
+
tags:
|
| 5 |
+
- image-classification
|
| 6 |
+
- keras
|
| 7 |
+
- cnn
|
| 8 |
+
- vehicles
|
| 9 |
+
- car-vs-bike
|
| 10 |
+
library_name: keras
|
| 11 |
+
model-index:
|
| 12 |
+
- name: Car vs Bike Classifier
|
| 13 |
+
results:
|
| 14 |
+
- task:
|
| 15 |
+
type: image-classification
|
| 16 |
+
name: Image Classification
|
| 17 |
+
metrics:
|
| 18 |
+
- name: Accuracy
|
| 19 |
+
type: accuracy
|
| 20 |
+
value: 0.96
|
| 21 |
+
---
|
| 22 |
+
|
| 23 |
# 🚗🏍️ Car vs Bike Image Classifier
|
| 24 |
|
| 25 |
This repository contains a Convolutional Neural Network (CNN) model trained to classify whether an image contains a **car** or a **bike**.
|
|
|
|
| 27 |
## 📊 Model Overview
|
| 28 |
|
| 29 |
- **Input:** RGB image resized to 128x128
|
| 30 |
+
- **Output:** Binary classification (0 = Bike, 1 = Car)
|
|
|
|
|
|
|
| 31 |
- **Framework:** TensorFlow / Keras
|
| 32 |
- **Model Format:** `.h5` (HDF5)
|
| 33 |
|
|
|
|
| 65 |
print("Prediction: 🚗 Car")
|
| 66 |
else:
|
| 67 |
print("Prediction: 🏍️ Bike")
|
| 68 |
+
```
|
| 69 |
|
| 70 |
+
## 📁 Files in This Repo
|
| 71 |
|
| 72 |
+
- `car_vs_bike_model.h5` → Trained model file
|
| 73 |
+
- `README.md` → Project description & usage
|
| 74 |
+
- (Optional) `example_images/` → Sample test images
|
| 75 |
|
| 76 |
+
## 📥 Download Instructions
|
| 77 |
|
| 78 |
+
Clone this repo and use Git LFS for large model files:
|
| 79 |
|
| 80 |
+
```bash
|
|
|
|
|
|
|
| 81 |
git lfs install
|
| 82 |
+
git clone https://huggingface.co/D-Neeraja/car_vs_bike_classifier
|
| 83 |
+
```
|
| 84 |
+
|
| 85 |
+
## 🧰 Requirements
|
| 86 |
|
| 87 |
+
- Python 3.7+
|
| 88 |
+
- TensorFlow ≥ 2.5
|
| 89 |
+
- NumPy
|
| 90 |
|
| 91 |
+
## 🙋♀️ Author
|
|
|
|
|
|
|
|
|
|
|
|
|
| 92 |
|
| 93 |
+
**Neeraja Dakkata**
|
| 94 |
+
📧 dakkataneeraja0031@gmail.com
|
| 95 |
+
🔗 GitHub: [github.com/D-Neeraja](https://github.com/D-Neeraja)
|
| 96 |
|
| 97 |
+
---
|
|
|
|
|
|
|
|
|
|
| 98 |
|
| 99 |
+
*Feel free to use this model for educational purposes or integrate it into your own projects.*
|