8000 GitHub - chhyyi/Restormer: restormer for AO
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

chhyyi/Restormer

 
 

Repository files navigation

Restormer for de-aberration task

I am testing restormer on the project aberration_sbcho (private) Mostly, it follows 'real-denoising task' of orginal repo. I am not plan to run it on the multi-gpu system.

See original README.md for further informations

Changes

  • It was submodule but now it is runnable independently
  • Made new dataset class for 3D-PSF abrr dataset for generation of dataset instead of on-the-fly use... cuz it was too slow

To do

  • prepare to run on remote server, via bash script file : Trained one model with patch size 384...(which causes oom in RTX 4090) It should be merged with local server version...

Usage

option file should be passed to the train/test scripts -opt (option)

python basicsr\train.py -opt Denoising\Options\multabrr_mpmneural.yml # for train
python basicsr\test_ch.py -opt Denoising\Options\multabrr_mpmneural_test.yml # for test. should modify config file to pass desired 'resume state'

config files

All the config files at Denoising/Options just same as original repo. See README.md for usage...

  • multabrr_mpmneural.yml: Default, train. should be passed to basicsr/train.py
  • multabrr_mpmneural_test.yml should be basicsr/test.py
  • multabrr_mpmneural_test_AUGdata.yml should be passed to basicsr/test_1stack.py as option. rename original vis_outputs, visualization directory on outputs(experiments) directory.
  • To process dataset synabrr_mpmneural_3dpsf_example.yml, synabrr_mpmneural_2dpsf_createdset.yml

About

restormer for AO

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 97.2%
  • MATLAB 2.3%
  • Shell 0.5%
0