This repository accompanies the paper AutoMergeNet: AutoML-based M-Source Satellite Data Fusion Evaluated with Atmospheric Case Studies
The code is structured as follows:
- The
automergenetfolder contains a pip-installable package with code for:- Training
- Model architectures
- Data loading (for carbon monoxide and methane)
- Experiment utility
- The
configfolder contains configuration files for the experiments - The
scriptsfolder contains scripts for:- Running and evaluating NAS (
nas_search.py) - Preparing/preprocessing methane data (
create_methane_dataset.py).
- Running and evaluating NAS (
- The
env.yamlfile contains the conda packages used in the experiments
This project runs with an MIT license. Please see the LICENSE file for more details.
Important
The env.yaml file was manually compiled. If you encounter any issues with the environment, please create a new conda environment and install the packages listed in the env.yaml file. We found that for pytorch cuda support specifically, you first need to install pytorch and only then install torchvision. If you attempt to install both at the same time, it might install the CPU version instead of the GPU version.
The paths to the datasets are optional arguments in the Dataset class definitions that default to "data/<dataset_name>". The datasets are not included in this repository. The datasets used in the experiments are:
Important
Update the paths (data_dir and annotations_path) in the Dataset class definitions to point to the correct location of the datasets on your machine (absolute path).
- Download the methane data
- Process the methane data using the
create_methane_dataset.pyscript. - Download the carbon monoxide data
- Normalise the carbon monoxide data
- Update the paths to datasets in the Dataset class definitions (
multi_modal_data.pyandco_data.py) - Update save paths in the configuration files
- Run the experiments using the scripts in the
scriptsfolder
Please cite the following paper if you use this code:
J. Wąsala et al., "AutoMergeNet: AutoML-based M-Source Satellite Data Fusion Evaluated with Atmospheric Case Studies," in IEEE Journal of Selected Topics in Applied Earth Observations and Remote Sensing, doi: 10.1109/JSTARS.2025.3621068.