This repository contains code for performing segmentation inference on satellite images using deep learning models.
This project is designed to perform segmentation inference on satellite images using deep learning models. The repository includes scripts for making predictions on satellite images and utilities for image format conversions. 📷🛰️
To perform inference on a new set of satellite images stored in the S3 Bucket:
📂 Path: projet-slums-detection/data-raw/PLEIADES/<dep>/<year>/
Before running inference, you must register the new images by linking them to their corresponding geometry polygons stored in the partitioned Parquet file:
📂 Path: projet-slums-detection/data-raw/PLEIADES/filename-to-polygons/
Run the following command:
python -m src.build_filename_to_polygonsOnce registered, you can run inference on these new images using the API:
python -m src.make_predictions_from_api --dep <dep> --year <year>All API-related code is in the app/ folder, built using FastAPI ⚡. The key files include:
- main.py 📌: Defines the API endpoints.
- utils.py 🔧: Contains utility functions for API operations.
The argo-workflows/ folder contains templates that enable automation and parallelization of inference across multiple departments and years. ⚡🔄
This project is licensed under the MIT License. 📄✅