🐟 Cyprus Fish Classifier (ConvNeXT Tiny)
This model is a fine-tuned version of facebook/convnext-tiny-224 on the Cyprus Fish Dataset. It is designed to classify fish species commonly found in the waters around Cyprus.
This model is part of a complete End-to-End MLOps project including CI/CD, Docker containerization, and a deployed FastAPI application.
- 💻 GitHub Repository: cyprus-fish-classifier
- 🚀 Live Demo: Hugging Face Space
📊 Model Details
- Model Architecture: ConvNext Tiny (28M parameters)
- Task: Image Classification (Multi-class)
- Input Image Size: 224x224 pixels
- Training Framework: PyTorch & Hugging Face Transformers
- Finetuning Strategy: Full finetuning of the classification head, backbone frozen
🚀 Get Started with the Model
Make sure your transformers>=4.57.3
from transformers import ConvNextImageProcessor,AutoImageProcessor,AutoModelForImageClassification
from PIL import Image
processor = AutoImageProcessor.from_pretrained("JayRay5/convnext-tiny-224-cyprus-fish-cls")
model = AutoModelForImageClassification.from_pretrained("JayRay5/convnext-tiny-224-cyprus-fish-cls")
image = Image.open("path_to_your_image/image.png").convert("RGB")
inputs = (
processor(images=image, return_tensors="pt").to(model.device).to(model.dtype)
)
with torch.inference_mode():
outputs = model(**inputs)
id2label = model.config.id2label
results = {}
for idx, prob in enumerate(probs):
idx_int = idx
label_name = id2label[idx_int]
results[label_name] = float(prob)
print(results)
Training Details
Training Data
Training data can be found at Cyprus Fish Dataset.
Training Procedure
The hyperparameters have been validated using a k-fold validation strategy.
For more details about the training process, refer to the GitHub repository.
Preprocessing
- Random Horizontal Flip (p=0.5)
- Random Rotation (±15 degrees)
- Color Jitter (Brightness & Contrast ±20%)
Training Hyperparameters
- k_folds: 5
- batch_size: 32
- grad_acc: 1
- epochs: 50
- lr: 3e-4
- scheduler: "constant"
- device: "cuda"
- fp16: True
- freeze_backbone: True
Evaluation
Factors
As the test set is small, the evaluation may be biased.
Metrics
Accuracy: 0.95
Out-of-Scope Use
The Cyprus Fish Dataset is a non-exhaustive dataset.
Thus, the model will not be able to classify species that are not in the dataset.