什么是即插即用?
即插即用是通过允许您灵活地更改正向模型和图像之前的方式,使在计算成像中优化图像恢复问题的方法更容易。
这是实现即插即用ADMM的尝试。
压缩感测
安装
pip install -r requirements.txt
您可以在此处下载DnCNN权重
用法
您需要3个对象来运行即插即用ADMM。
前向模型
图像优先
变量变换
变量转换是变量x和v之间的转换,其中x是正向模型的输入自变量,而v是先验模型的输入自变量。 该优化将运行约束优化的增强拉格朗日变换,transform(x)= v。
您可以编写自己的正向模型类和图像优先类,但是它必须具有以下属性和方法
class your_forward_or_prior_model :
def __init__ ( self , input_shape ):
self . input_shape = input_
2021-09-04 01:02:32
16.92MB
Python
1