# AE_fusion

**Repository Path**: pengby20/AE_fusion

## Basic Information

- **Project Name**: AE_fusion
- **Description**: 多模态融合训练和测试相关代码
- **Primary Language**: Python
- **License**: Not specified
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 2
- **Forks**: 1
- **Created**: 2023-02-03
- **Last Updated**: 2025-10-31

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

<!--
 * @Author: Peng Bo
 * @Date: 2023-02-02 08:25:04
 * @LastEditTime: 2023-02-15 18:21:04
 * @Description: 
 * 
-->
## visible-infrared fusion and image dehazing by deep learning

### fusion
#### prepare the dataset
1. download MSRS, and extract
2. process the data into h5py format, `python utils/dataprocessing.py`

#### training
1. training: `python train.py --config-file your_config_path`
  default `your_config_path` is set as `experiments/template/config.yaml`, 
`experiments/restormer_coa_fusion/config.yaml` shows how to configure it.

#### testing
1. test demo: `python train.py --config-file your_config_path --eval`: 

### image dehazing
1. test demo: `python models/aod_net.py`
2. training: coming soon.