能量控制问题代码matlab
这是通过[1]中介绍的多级方法进行的最优运输(OT)的Matlab代码存储库。
OT在许多领域都发挥着至关重要的作用,包括流体动力学,图像处理,机器学习和控制。
它是两个概率分布的适当距离度量。
该距离也称为地球移动者距离(EMD)或Wasserstein距离。
我们的Matlab代码可计算网格上定义的两个分布之间的Wasserstein-1距离。
它使用以下三个基本度量之一:1-范数,2-范数和inf-范数。
我们的代码具有以下优点:
快速地。
在单个CPU上花费几秒钟来计算1024
*
1024网格上的最佳传输。
方便使用的。
问题描述
给定两个2D概率分布或两个图像rho0和rho1,我们希望找到能量最小的从一个到另一个的传输:
在这里,p可以是1、2或无穷大,m(x)的1、2或无穷大范数分别是,h是网格步长。
算法与函数
“
otFunctions
/
W1PD_ML.m”在[1]中实现算法1M;
“
otFunctions
/
W1PDHG_ML.m”在[1]中实现了算法2M。
句法
[m,phi]
=
W1PD_ML(h,
rho0,
rho1
2022-11-15 14:40:41
1.48MB
系统开源
1