Edit model card

Model Details

This model is based on the ResNet-18 architecture and it has been fine-tuned for satellite image classification tasks on the EuroSAT dataset.

Model type: Convolutional Neural Network (CNN)

Finetuned from model : ResNet18 (pre-trained on ImageNet-1k)

Model Sources

Repository: https://github.com/chathumal93/EuroSat-RGB-Classifiers

Training Details

Training Data

The dataset comprises JPEG composite chips extracted from Sentinel-2 satellite imagery, representing the Red, Green, and Blue bands. It encompasses 27,000 labeled and geo-referenced images across 10 Land Use and Land Cover (LULC) classes

Training Procedure

Preprocessing: Standard image preprocessing including resizing, center cropping, normalization, and data augmentation techniques [RandomHorizontalFlip and RandomVerticalFlip]

Training Hyperparameters

  • Learning rate: 3e-5
  • Batch size: 64
  • Optimizer: AdamW
  • Scheduler: PolynomialLR
  • Loss: CrossEntropyLoss
  • Betas=(0.9, 0.999)
  • Weight_decay=0.01
  • Epochs: 20

Evaluation

Results

Model Phase Avg Loss Accuracy
resnet18-eurosat Train 0.097586 97.01%
Validation 0.071375 97.70%
Test 0.068443 97.74%
Model Accuracy Precision Recall F1
resnet18-eurosat 97.74% 0.97747 0.97741 0.97740
Downloads last month
4
Safetensors
Model size
11.2M params
Tensor type
F32
·
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

Dataset used to train cm93/resnet18-eurosat