search

caiyuanhao1998 / MST

A toolbox for spectral compressive imaging reconstruction including MST (CVPR 2022), CST (ECCV 2022), DAUHST (NeurIPS 2022), BiSCI (NeurIPS 2023), HDNet (CVPR 2022), MST++ (CVPRW 2022), etc.
MIT License
571 stars 71 forks source link

Question about the ADMM-Net #52

Closed 050644zf closed 5 months ago

050644zf commented 5 months ago

I found some differences in the original ADMM-net and your implementation

  1. Original one has sparse inverse operation to achieve 2nd equation of formula (12) in original paper, while yours doesn't. Is there any change applied on formulas?
  2. Your implementation has multiple stages of U-Net which doesn't appear in original paper. Instead, I found this paper has similar usage of U-Net as denoising network here. Did you use U-Net as some kind of equivalent to part of the original ADMM-net or sth else?
caiyuanhao1998 commented 5 months ago

  1. We just change the unfolding scheme while keeping others the same.

  2. We use multiple stages of Unets for fair comparison

caiyuanhao1998 commented 5 months ago

If you find this repo helpful, please help us star it. Thanks