Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,16 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
# 🐄 Breed-Recognizer
|
| 2 |
|
| 3 |
A deep learning-based cattle breed recognition system using PyTorch and ConvNeXt. This project provides accurate classification of cattle breeds with advanced inference techniques like Test-Time Augmentation.
|
|
@@ -57,7 +70,7 @@ from classifier import BreedPredictor
|
|
| 57 |
# Initialize the predictor
|
| 58 |
predictor = BreedPredictor(
|
| 59 |
model_path="best_breed_model.pth",
|
| 60 |
-
num_classes=
|
| 61 |
class_names=["Gir", "Sahiwal", "Kankrej", "Breed4", ...]
|
| 62 |
)
|
| 63 |
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language: en
|
| 3 |
+
tags:
|
| 4 |
+
- image-classification
|
| 5 |
+
- computer-vision
|
| 6 |
+
- pytorch
|
| 7 |
+
- convnext
|
| 8 |
+
- cattle-breed-recognition
|
| 9 |
+
license: mit
|
| 10 |
+
metrics:
|
| 11 |
+
- accuracy
|
| 12 |
+
---
|
| 13 |
+
|
| 14 |
# 🐄 Breed-Recognizer
|
| 15 |
|
| 16 |
A deep learning-based cattle breed recognition system using PyTorch and ConvNeXt. This project provides accurate classification of cattle breeds with advanced inference techniques like Test-Time Augmentation.
|
|
|
|
| 70 |
# Initialize the predictor
|
| 71 |
predictor = BreedPredictor(
|
| 72 |
model_path="best_breed_model.pth",
|
| 73 |
+
num_classes=26,
|
| 74 |
class_names=["Gir", "Sahiwal", "Kankrej", "Breed4", ...]
|
| 75 |
)
|
| 76 |
|