| # EfficientNetB0 Fine-tuned Model |
|
|
| This repository contains a fine-tuned EfficientNetB0 model for image classification. |
|
|
| ## Model Information |
|
|
| - **Architecture**: EfficientNetB0 |
| - **Training Date**: 2025-05-17 |
| - **Dataset**: Custom dataset for image classification |
| - **Input Size**: 224x224 |
| - **Number of Classes**: 14 |
|
|
| ## Usage |
|
|
| You can use this model in your code like this: |
|
|
| ```python |
| from transformers import EfficientNetForImageClassification, EfficientNetFeatureExtractor |
| import torch |
| |
| # Load model and feature extractor |
| model = EfficientNetForImageClassification.from_pretrained("Nafi007/EfficientNetB0") |
| feature_extractor = EfficientNetFeatureExtractor.from_pretrained("Nafi007/EfficientNetB0") |
| |
| # Process image and make prediction |
| inputs = feature_extractor(images=image, return_tensors="pt") |
| outputs = model(**inputs) |
| predictions = torch.nn.functional.softmax(outputs.logits, dim=-1) |
| ``` |
|
|
| ## Citation |
|
|
| If you use this model in your research, please cite it as: |
|
|
| ```bibtex |
| @misc{efficientnetb0_2025, |
| title = {EfficientNetB0 Fine-tuned Model}, |
| author = {Nafi007}, |
| year = {2025}, |
| publisher = {Hugging Face} |
| } |
| ``` |
|
|
