Vishu2006 commited on
Commit
1e9be3c
·
verified ·
1 Parent(s): edba9de

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -1
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=12,
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