| # 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} | |
| } | |
| ``` | |