Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
ChengShen 73112b185e | 2 years ago | |
---|---|---|
Fig | 3 years ago | |
.gitignore | 3 years ago | |
LICENSE | 3 years ago | |
README.md | 3 years ago | |
dataset.py | 2 years ago | |
model.py | 3 years ago | |
prepare_data.py | 3 years ago | |
test.py | 3 years ago | |
train_mge.py | 3 years ago | |
utils.py | 3 years ago |
Code for CVPR21 paper NBNet.
The illustration of our key insight:
python prepare_data.py --data_dir yours_sidd_data_path
For SIDD benchmark, use:
python train_mge.py -d prepared_data_path -n num_gpus
For DnD benchmark, we use MixUp additionally:
python train_mge.py -d prepared_data_path -n num_gpus --dnd
Download the pretrained checkpoint and use:
python test.py -d prepared_data_path -c checkpoint_path
The result is PSNR 39.765.
MegEngine checkpoint for SIDD benchmark can be downloaded via
Google Drive
or
GitHub Release.
No Description
Python
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》