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:

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:

@misc{efficientnetb0_2025,
    title = {EfficientNetB0 Fine-tuned Model},
    author = {Nafi007},
    year = {2025},
    publisher = {Hugging Face}
}