Skip to content

LED V0.1.1 Release

Latest
Compare
Choose a tag to compare
@Srameo Srameo released this 13 Jan 08:15
· 6 commits to main since this release

This release provides LED models that has been pre-trained and fine-tuned on more powerful network structures, namely Restormer and NAFNet.

Explanation on the pretrained models

The data naming convention consists of the following components: {Method}_{Phase}_{Dataset}_{Camera Model}_{Training Setting}_Setting_Ratio{Range}.

For the LED method, there are two phases in total: Pretrain and Deploy. For Pretrain, the checkpoint obtained cannot be directly tested on any dataset; it is only used for subsequent fine-tuning. On the other hand, for all methods, the Deploy means that the checkpoint contains parameters consistent with the UNet used in the SID method, making it directly suitable for testing.

Regarding the training setting, there are two mainstream settings known as ELD (CVPR20) and PMN (MM22) settings. We represent these two settings as "CVPR20" and "MM22," respectively.

e.g. LED+NAFNet_Deploy_SID_SonyA7S2_CVPR20_Setting_Ratio100-300:

  1. "LED+NAFNet_Deploy": This refers to the LED method in the "deploy" phase. And with the NAFNet architecture.
  2. "SID_SonyA7S2": This indicates that the testing is done on the SonyA7S2 subset of the SID dataset.
  3. "CVPR20_Setting": This means that the training strategy during the "pretrain" phase is the same as the one used in the "ELD (CVPR20)" setting.
  4. "Ratio100-300": This indicates the range of the ratio is from 100 to 300.