This repository is for educational purposes only. Any use of this code, dataset, or methodology for research, commercial, or academic purposes is strictly prohibited without prior written consent from the authors.
This repository contains the implementation, datasets (hosted on Hugging Face), and results for the paper "A Hybrid Random Forest and CNN Framework for Tile-Wise Oil-Water Classification in Hyperspectral Images", authored by Mehdi Nickzamir and S. Mohammad Sheikh Ahmadi.
Oil spills are devastating for the environment and human activities, requiring timely detection to mitigate their impacts. This project proposes a hybrid framework that combines the strengths of Random Forest (RF) for pixel-wise classification and Convolutional Neural Networks (CNN) for incorporating spatial context to enhance oil-water classification in hyperspectral images (HSI).
- Hyperspectral Oil Spill Detection (HOSD) dataset preprocessing pipeline.
- Implementation of the hybrid RF+CNN framework.
- Comparison with baseline and state-of-the-art models.
- Pretrained models and metrics for evaluation.
| File/Folder | Description |
|---|---|
train.ipynb |
Training script for the hybrid RF+CNN model. |
test.ipynb |
Testing script for evaluating the model. |
preprocessing.ipynb |
Preprocessing pipeline including noisy channel removal, normalization, PCA, tiling, and augmentation. |
ADSP.pdf |
Full research paper detailing the methodology, experiments, and results. |
The datasets and models are hosted on Hugging Face. Links to the recourses:
These datasets include:
- Original Dataset: Raw HOSD dataset with 18 hyperspectral images.
- Preprocessed Dataset: Dataset after noisy channel removal, normalization, PCA, and tiling.