MCPcopy Index your code
hub / github.com/294coder/Dif-PAN

github.com/294coder/Dif-PAN @main

Chat with this repo
repository ↗ · DeepWiki ↗ · + Follow
363 symbols 927 edges 18 files 54 documented · 15%
What it actually does AI analysis from the code graph — generated when you open this
loading…
README

DDIF: Diffusion model with disentangled modulations for sharpening multispectral and hyperspectral images

ArXiv Arxiv Elsevier logo Information Fusion

Abstract: The denoising diffusion model has received increasing attention in the field of image generation in recent years, thanks to its powerful generation capability. However, diffusion models should be deeply investigated in the field of multi-source image fusion, such as remote sensing pansharpening and multispectral and hyperspectral image fusion (MHIF). In this paper, we introduce a novel {supervised} diffusion model with two conditional modulation modules, specifically designed for the task of multi-source image fusion. These modules mainly consist of a coarse-grained style modulation (CSM) and a fine-grained wavelet modulation (FWM), which aim to disentangle coarse-grained style information and fine-grained frequency information, respectively, thereby generating competitive fused images. Moreover, some essential strategies for the training of the given diffusion model are well discussed, e.g., the selection of training objectives. The superiority of the proposed method is verified compared with recent state-of-the-art (SOTA) techniques by extensive experiments on two multi-source image fusion benchmarks, i.e., pansharpening and MHIF. In addition, sufficient discussions and ablation studies in the experiments are involved to demonstrate the effectiveness of our approach.

News: - 2024/4/21: A new paper about Neural Shrödinger Bridge Matching for Pansharpening is released on arxiv.🤗

  • 2023/12/4:Code RELEASED!:fire:

  • 2023/11/23: Code will be released soon!:fire:

Quick Overview

The code in this repo supports Pansharpening, Hyperspectral and multispectral image fusion.

Instructions

Dataset

In this office repo, you can find the Pansharpening dataset of WV3, GF2, and QB.

We follow the PSRT to implement Hyperspectral and multispectral image fusion. You can find the data we use in this repo.

Other instructions will come soon!

Citation

If you find our paper useful, please consider citing the following:

@article{DDIF,
title = {Diffusion model with disentangled modulations for sharpening multispectral and hyperspectral images},
journal = {Information Fusion},
volume = {104},
pages = {102158},
year = {2024},
issn = {1566-2535},
doi = {https://doi.org/10.1016/j.inffus.2023.102158},
url = {https://www.sciencedirect.com/science/article/pii/S1566253523004748},
author = {Zihan Cao and Shiqi Cao and Liang-Jian Deng and Xiao Wu and Junming Hou and Gemine Vivone},
}

Core symbols most depended-on inside this repo

extract
called by 23
diffusion/diffusion_ddpm_pan.py
marginal_std
called by 19
solver/dpm_solver.py
marginal_lambda
called by 19
solver/dpm_solver.py
log
called by 18
utils/logger.py
marginal_log_mean_coeff
called by 16
solver/dpm_solver.py
model_fn
called by 15
solver/dpm_solver.py
print
called by 11
utils/logger.py
exist
called by 8
utils/misc.py

Shape

Method 207
Function 88
Class 68

Languages

Python100%

Modules by API surface

models/sr3_dwt.py49 symbols
models/sr3.py48 symbols
models/unet_model_google.py37 symbols
diffusion/diffusion_ddpm_pan.py37 symbols
solver/dpm_solver.py34 symbols
utils/loss_utils.py24 symbols
utils/metric.py19 symbols
utils/misc.py16 symbols
utils/lr_scheduler.py15 symbols
utils/logger.py14 symbols
models/pansharpen_model.py14 symbols
dataset/pan_dataset.py14 symbols

For agents

$ claude mcp add Dif-PAN \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact