BiliSakura commited on
Commit
a304e08
·
verified ·
1 Parent(s): 844428c

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +53 -0
README.md ADDED
@@ -0,0 +1,53 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - image-classification
5
+ - remote-sensing
6
+ - resnet
7
+ - pytorch
8
+ - transformers
9
+ ---
10
+
11
+ # RSP-ResNet-50
12
+
13
+ ResNet-50 based model for remote sensing scene classification (51 classes).
14
+
15
+ ## Usage
16
+
17
+ ```python
18
+ from transformers import AutoModelForImageClassification
19
+ import torch
20
+
21
+ # Load model
22
+ model = AutoModelForImageClassification.from_pretrained(
23
+ "BiliSakura/RSP-ResNet-50",
24
+ trust_remote_code=True
25
+ )
26
+
27
+ # Inference
28
+ model.eval()
29
+ input_image = torch.randn(1, 3, 224, 224) # (batch, channels, height, width)
30
+
31
+ with torch.no_grad():
32
+ outputs = model(pixel_values=input_image)
33
+ logits = outputs.logits # Shape: (1, 51)
34
+ predicted_class = logits.argmax(dim=-1).item()
35
+ ```
36
+
37
+ ## Model Details
38
+
39
+ - **Architecture:** ResNet-50
40
+ - **Input size:** 224×224×3
41
+ - **Number of classes:** 51
42
+ - **Parameters:** ~23.6M
43
+
44
+ ## Citation
45
+
46
+ ```bibtex
47
+ @misc{rsp-resnet-50,
48
+ title={RSP-ResNet-50: Remote Sensing Scene Classification Model},
49
+ author={BiliSakura},
50
+ year={2024},
51
+ howpublished={\url{https://huggingface.co/BiliSakura/RSP-ResNet-50}}
52
+ }
53
+ ```