sahilvishwa2108's picture
Deploy optimized lightweight oil spill detection API - Runtime model download, memory efficient
ada534f
metadata
title: Oil Spill Detection API
emoji: 🛢️
colorFrom: blue
colorTo: green
sdk: docker
pinned: false
license: mit
app_port: 7860

Oil Spill Detection API

A lightweight FastAPI application for detecting oil spills in satellite imagery using deep learning models.

Features

  • U-Net Model: Semantic segmentation for oil spill detection
  • DeepLab V3+ Model: Advanced semantic segmentation
  • Lazy Loading: Models load on-demand to save memory
  • RESTful API: Easy integration with frontend applications

Usage

Health Check

GET /health

Predict Oil Spill

POST /predict
Content-Type: multipart/form-data

Parameters:
- file: Image file (JPG, PNG)
- model_choice: "model1" (U-Net) or "model2" (DeepLab)

API Documentation

Visit /docs for interactive API documentation.

Model Information

  • Model 1: U-Net architecture for semantic segmentation
  • Model 2: DeepLab V3+ for enhanced accuracy
  • Input Size: 256x256 pixels
  • Output: Binary classification (Oil Spill / No Oil Spill)

This API is optimized for HuggingFace Spaces with memory-efficient model loading.