cm93 commited on
Commit
d6c1a56
1 Parent(s): 5d453ee

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +44 -2
README.md CHANGED
@@ -3,6 +3,48 @@ tags:
3
  - image-classification
4
  - timm
5
  library_name: timm
6
- license: apache-2.0
 
 
7
  ---
8
- # Model card for resnet50-eurosat
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  - image-classification
4
  - timm
5
  library_name: timm
6
+ datasets:
7
+ - cm93/eurosat
8
+ pipeline_tag: image-classification
9
  ---
10
+ ## Model Details
11
+
12
+ **Model type:** Convolutional Neural Network (CNN)
13
+
14
+ **Finetuned from model :** ResNet50 (pre-trained on ImageNet-1k at resolution 224x224)
15
+
16
+ ### Model Sources
17
+ **Repository:** https://github.com/chathumal93/EuroSat-RGB-Classifiers
18
+
19
+ ## Training Details
20
+
21
+ ### Training Data
22
+
23
+ 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
24
+
25
+ ### Training Procedure
26
+ **Preprocessing:** Standard image preprocessing including resizing, center cropping, normalization, and data augmentation techniques [RandomHorizontalFlip and RandomVerticalFlip]
27
+
28
+ ### Training Hyperparameters
29
+ - **Learning rate:** 3e-5
30
+ - **Batch size:** 64
31
+ - **Optimizer:** AdamW
32
+ - **Scheduler:** PolynomialLR
33
+ - **Loss:** CrossEntropyLoss
34
+ - **Betas**=(0.9, 0.999)
35
+ - **Weight_decay**=0.01
36
+ - **Epochs:** 20
37
+
38
+
39
+ ## Evaluation
40
+
41
+ ### Results
42
+ | Model | Phase | Avg Loss | Accuracy |
43
+ |:----------------------------:|:----------:|:--------:|:---------:|
44
+ | resnet50-eurosat | Train | 0.076420 | 97.56% |
45
+ | | Validation | 0.054377 | 98.30% |
46
+ | | Test | 0.058930 | 98.07% |
47
+
48
+ | Model | Accuracy | Precision | Recall | F1 |
49
+ |:----------------------------:|:--------:|:------------:|:-----------:|:-----------:|
50
+ | resnet50-eurosat | 98.07% | 0.98078 | 0.98074 | 0.98074 |