├── .gitignore ├── LICENSE ├── README.md ├── ckpt └── .gitkeep ├── conf ├── APBSN_DND.yaml ├── APBSN_NIND.yaml ├── APBSN_SIDD.yaml └── APBSN_SIDDbench.yaml ├── figs ├── main.png ├── results.png └── visual_results.png ├── prep.py ├── src ├── datahandler │ ├── DND.py │ ├── NIND.py │ ├── SIDD.py │ ├── __init__.py │ ├── custom.py │ ├── denoise_dataset.py │ └── prepare_dataset.md ├── loss │ ├── __init__.py │ ├── recon.py │ └── recon_self.py ├── model │ ├── APBSN.py │ ├── DBSNl.py │ └── __init__.py ├── trainer │ ├── __init__.py │ ├── base.py │ └── trainer.py └── util │ ├── config_parse.py │ ├── dnd_submission │ ├── bundle_submissions.py │ ├── dnd_denoise.py │ └── pytorch_wrapper.py │ ├── file_manager.py │ ├── logger.py │ ├── progress_msg.py │ ├── summary_logging.py │ └── util.py ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/README.md -------------------------------------------------------------------------------- /ckpt/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /conf/APBSN_DND.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/conf/APBSN_DND.yaml -------------------------------------------------------------------------------- /conf/APBSN_NIND.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/conf/APBSN_NIND.yaml -------------------------------------------------------------------------------- /conf/APBSN_SIDD.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/conf/APBSN_SIDD.yaml -------------------------------------------------------------------------------- /conf/APBSN_SIDDbench.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/conf/APBSN_SIDDbench.yaml -------------------------------------------------------------------------------- /figs/main.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/figs/main.png -------------------------------------------------------------------------------- /figs/results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/figs/results.png -------------------------------------------------------------------------------- /figs/visual_results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/figs/visual_results.png -------------------------------------------------------------------------------- /prep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/prep.py -------------------------------------------------------------------------------- /src/datahandler/DND.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/datahandler/DND.py -------------------------------------------------------------------------------- /src/datahandler/NIND.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/datahandler/NIND.py -------------------------------------------------------------------------------- /src/datahandler/SIDD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/datahandler/SIDD.py -------------------------------------------------------------------------------- /src/datahandler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/datahandler/__init__.py -------------------------------------------------------------------------------- /src/datahandler/custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/datahandler/custom.py -------------------------------------------------------------------------------- /src/datahandler/denoise_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/datahandler/denoise_dataset.py -------------------------------------------------------------------------------- /src/datahandler/prepare_dataset.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/datahandler/prepare_dataset.md -------------------------------------------------------------------------------- /src/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/loss/__init__.py -------------------------------------------------------------------------------- /src/loss/recon.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/loss/recon.py -------------------------------------------------------------------------------- /src/loss/recon_self.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/loss/recon_self.py -------------------------------------------------------------------------------- /src/model/APBSN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/model/APBSN.py -------------------------------------------------------------------------------- /src/model/DBSNl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/model/DBSNl.py -------------------------------------------------------------------------------- /src/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/model/__init__.py -------------------------------------------------------------------------------- /src/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/trainer/__init__.py -------------------------------------------------------------------------------- /src/trainer/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/trainer/base.py -------------------------------------------------------------------------------- /src/trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/trainer/trainer.py -------------------------------------------------------------------------------- /src/util/config_parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/util/config_parse.py -------------------------------------------------------------------------------- /src/util/dnd_submission/bundle_submissions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/util/dnd_submission/bundle_submissions.py -------------------------------------------------------------------------------- /src/util/dnd_submission/dnd_denoise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/util/dnd_submission/dnd_denoise.py -------------------------------------------------------------------------------- /src/util/dnd_submission/pytorch_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/util/dnd_submission/pytorch_wrapper.py -------------------------------------------------------------------------------- /src/util/file_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/util/file_manager.py -------------------------------------------------------------------------------- /src/util/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/util/logger.py -------------------------------------------------------------------------------- /src/util/progress_msg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/util/progress_msg.py -------------------------------------------------------------------------------- /src/util/summary_logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/util/summary_logging.py -------------------------------------------------------------------------------- /src/util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/src/util/util.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wooseoklee4/AP-BSN/HEAD/train.py --------------------------------------------------------------------------------