Thanks to visit codestin.com
Credit goes to github.com

Skip to content

DLR-MF-DAS/SmallMinesDS

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SmallMinesDS

A Multi-Modal Dataset for Mapping Artisanal and Small-Scale Gold Mines

[Stella Ofori-Ampofo]1, [Antony Zappacosta]2, [Ridvan Salih Kuzu]2, [Peter Schauer]3, [Martin Willberg]3, [Xiao Xiang Zhu]1,4,

1 Technical University of Munich, 2 German Aerospace Center (DLR), 3 Industrieanlagen-Betriebsgesellschaft mbH (IABG), 4 Munich Center for Machine Learning

Paper: Journal (DOI: 10.1109/LGRS.2025.3566356)

Updates

Jan. 10, 2025 The first version of SmallMinesDS is available on HuggingFace

Introduction

The gradual expansion of unregularized artisanal small-scale gold mining (ASGM) fuels environmental degradation and poses a risk to miners and mining communities. To enforce sustainable mining, support reclamation initiatives, and pave the way for understudying the impacts of mining, we present SmallMinesDS, a dataset for mapping artisanal small-scale gold mining from multi-sensor satellite images. The initial version of the dataset covers five districts in Southwestern Ghana in two time periods. The first version of SmallMinesDS covers five administrative districts in Southwestern Ghana with a ground area of about 3200 sq.km. SmallMinesDS includes satellite images from optical and radar sensors in the dry season (January) for two years (2016 and 2022). The area covered is ecologically diverse, hosting rich biodiversity and serving as a hub for multiple land-use activities, including crop cultivation, forestry, industrial and small-scale gold mining. This multifaceted landscape underscores the importance of reliable mapping and monitoring of ASGM activities.

area of interest

Data description

  • SmallMinesDS contain 4270 patches; 2175 each for 2016 and 2022

  • Each patch has an input shape of 13 x 128 x 128 and a corresponding mask of 1 x 128 x 128

  • The open-access data can be downloaded at Hugging Face

sample images

band order

Quick start

Prerequisites

  • Conda (for managing environments)
  • NVIDIA GPU (for efficient computation)

Setting Up Environments

Two separate virtual environments are required:

  1. Terratorch Environment (for TerraTorch dependencies)
  2. SAM2 Environment (for Segment Anything Model v2 dependencies)

These environments are necessary as they rely on different PyTorch versions.

Creating the TerraTorch Environment

conda create -n terratorch python=3.11
conda activate terratorch
pip install -r requirements.txt
conda deactivate

Creating the SAM2 Environment

conda create -n sam2 python=3.11
conda activate sam2
chmod +x install_sam2.sh
bash install_sam2.sh
mv scripts/ft-sam2.py sam2/
conda deactivate

Fine-tuning Prithvi-2

Fine-tune Prithvi-2 using our dataset.

Fine-tuning with the 300M Model
conda activate terratorch
python scripts/train-prithvi-v2-300.py
Fine-tuning with the 600M Model
conda activate terratorch
python scripts/train-prithvi-v2-600.py

Training ResNet50 from Scratch

To compare with Prithvi-2, we train ResNet50 from scratch using six spectral bands: Blue, Green, Red, Narrow NIR, SWIR, and SWIR 2.

conda activate terratorch
python scripts/train-resnet50-6bands.py

Fine-tuning the Segment Anything Model v2 (SAM2)

Segment Anything Model v2 is a foundational model designed for promptable visual segmentation in RGB images and videos. We fine-tune it using the RGB channels of our dataset.

conda activate sam2
python sam2/ft-sam2.py

Fine-tuning ResNet50 Pretrained on ImageNet

For a fair comparison in the RGB domain, we fine-tune a ResNet50 model pretrained on ImageNet and compare its performance with SAM2.

conda activate terratorch
python scripts/ft-resnet50.py

Citation

If you use the dataset or supporting code in your research, please cite SmallMinesDS as :

@ARTICLE{10982207,
  author={Ofori-Ampofo, Stella and Zappacosta, Antony and Kuzu, Rıdvan Salih and Schauer, Peter and Willberg, Martin and Zhu, Xiao Xiang},
  journal={IEEE Geoscience and Remote Sensing Letters}, 
  title={SmallMinesDS: A Multi-Modal Dataset for Mapping Artisanal and Small-Scale Gold Mines}, 
  year={2025},
  volume={},
  number={},
  pages={1-1},
  keywords={Data mining;Foundation models;Gold;Satellite images;Sentinel-1;Optical sensors;Laser radar;Biological system modeling;Spaceborne radar;Radar imaging;Earth Observation;Semantic Segmentation;Mining;Machine Learning;Foundation Models},
  doi={10.1109/LGRS.2025.3566356}}


About

Benchmarking foundation models for semantic segmentation of small scale mining activities

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •