·
24 commits
to main
since this release
Model weights for the MIRNet implemented in this repository.
The model has been trained for 100 epochs on the LoL dataset scaled down to 64x64 images (based on the weights released here), and then fine-tuned for 100 epochs on the custom dataset released here, with early stopping intervening at 35 epochs.
It and obtains a PSNR of 19.64 on the custom-dataset test data, and increase of 1.24 in comparison to the PSNR of 18.4 obtained by the pre-trained model. With the fine-tuned model, the enhanced images contain less noises, and have colors that better correspond to the ground truth.
The performance on the custom test data can be visualized in the following figure:
