VOOZH about

URL: https://huggingface.co/prithivMLmods/RSI-CB256-07

⇱ prithivMLmods/RSI-CB256-07 · Hugging Face


👁 4.png

RSI-CB256-07

RSI-CB256-07 is a SigLIP2-based model fine-tuned for coarse-grained remote sensing land-cover classification. It distinguishes among 7 essential categories commonly used in environmental, urban planning, and geospatial analysis applications. The model is built on google/siglip2-base-patch16-224 using the SiglipForImageClassification architecture.

Classification Report:
 precision recall f1-score support

 transportation 0.9810 0.9858 0.9834 3300
 other objects 0.9854 0.9932 0.9893 884
 woodland 0.9973 0.9958 0.9966 6258
 water area 0.9870 0.9837 0.9854 4104
 other land 0.9925 0.9919 0.9922 3593
 cultivated land 0.9918 0.9901 0.9909 2817
construction land 0.9945 0.9963 0.9954 3791

 accuracy 0.9912 24747
 macro avg 0.9899 0.9910 0.9904 24747
 weighted avg 0.9912 0.9912 0.9912 24747

👁 download.png


Label Space: 7 Remote Sensing Classes

This model predicts one of the following categories for a given satellite or aerial image:

Class 0: "transportation" 
Class 1: "other objects" 
Class 2: "woodland" 
Class 3: "water area" 
Class 4: "other land" 
Class 5: "cultivated land" 
Class 6: "construction land"

Install Dependencies

pip install -q transformers torch pillow gradio

Inference Code

import gradio as gr
from transformers import AutoImageProcessor, SiglipForImageClassification
from PIL import Image
import torch

# Load model and processor
model_name = "prithivMLmods/RSI-CB256-07"
model = SiglipForImageClassification.from_pretrained(model_name)
processor = AutoImageProcessor.from_pretrained(model_name)

# ID to label mapping
id2label = {
 "0": "transportation",
 "1": "other objects",
 "2": "woodland",
 "3": "water area",
 "4": "other land",
 "5": "cultivated land",
 "6": "construction land"
}

def classify_rsi_image(image):
 image = Image.fromarray(image).convert("RGB")
 inputs = processor(images=image, return_tensors="pt")

 with torch.no_grad():
 outputs = model(**inputs)
 logits = outputs.logits
 probs = torch.nn.functional.softmax(logits, dim=1).squeeze().tolist()

 prediction = {
 id2label[str(i)]: round(probs[i], 3) for i in range(len(probs))
 }

 return prediction

# Gradio Interface
iface = gr.Interface(
 fn=classify_rsi_image,
 inputs=gr.Image(type="numpy"),
 outputs=gr.Label(num_top_classes=7, label="Predicted Land-Cover Category"),
 title="RSI-CB256-07",
 description="Upload a satellite or aerial image to classify it into one of seven coarse land-cover classes using SigLIP2."
)

if __name__ == "__main__":
 iface.launch()

Applications

  • Urban vs Rural Segmentation
  • Land-Use Classification
  • National/Regional Land Cover Monitoring
  • Environmental Impact Assessment
Downloads last month
5
Safetensors
Model size
92.9M params
Tensor type
F32
·

Model tree for prithivMLmods/RSI-CB256-07

Finetuned
(119)
this model

Dataset used to train prithivMLmods/RSI-CB256-07

Collection including prithivMLmods/RSI-CB256-07