├── .gitignore ├── LICENSE ├── README.md ├── config.py ├── data ├── test │ ├── mild_gt │ │ ├── 1_gt.png │ │ ├── 2_gt.png │ │ ├── 3_gt.png │ │ └── 4_gt.png │ ├── mild_in │ │ ├── 1_in.png │ │ ├── 2_in.png │ │ ├── 3_in.png │ │ └── 4_in.png │ ├── mine │ │ ├── 1.jpg │ │ ├── 2.jpg │ │ ├── 3.jpg │ │ └── 4.jpg │ ├── moderate_gt │ │ ├── 1_gt.png │ │ ├── 2_gt.png │ │ ├── 3_gt.png │ │ └── 4_gt.png │ ├── moderate_in │ │ ├── 1_in.png │ │ ├── 2_in.png │ │ ├── 3_in.png │ │ └── 4_in.png │ ├── severe_gt │ │ ├── 1_gt.png │ │ ├── 2_gt.png │ │ ├── 3_gt.png │ │ └── 4_gt.png │ └── severe_in │ │ ├── 1.png │ │ ├── 2.png │ │ ├── 3_in.png │ │ └── 4_in.png ├── train │ ├── generate_train.m │ ├── noise_combination.m │ └── store2hdf5.m └── valid │ └── validation.h5 ├── dqn ├── __init__.py ├── agent.py ├── base.py ├── environment.py ├── ops.py ├── replay_memory.py └── utils.py ├── imgs ├── framework.png ├── real_world.png ├── restore.gif └── tensorboard.png ├── main.py ├── models ├── checkpoint ├── model.data-00000-of-00001 ├── model.index └── model.meta └── toolbox ├── checkpoint ├── tool01.data-00000-of-00001 ├── tool01.index ├── tool01.meta ├── tool02.data-00000-of-00001 ├── tool02.index ├── tool02.meta ├── tool03.data-00000-of-00001 ├── tool03.index ├── tool03.meta ├── tool04.data-00000-of-00001 ├── tool04.index ├── tool04.meta ├── tool05.data-00000-of-00001 ├── tool05.index ├── tool05.meta ├── tool06.data-00000-of-00001 ├── tool06.index ├── tool06.meta ├── tool07.data-00000-of-00001 ├── tool07.index ├── tool07.meta ├── tool08.data-00000-of-00001 ├── tool08.index ├── tool08.meta ├── tool09.data-00000-of-00001 ├── tool09.index ├── tool09.meta ├── tool10.data-00000-of-00001 ├── tool10.index ├── tool10.meta ├── tool11.data-00000-of-00001 ├── tool11.index ├── tool11.meta ├── tool12.data-00000-of-00001 ├── tool12.index └── tool12.meta /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/config.py -------------------------------------------------------------------------------- /data/test/mild_gt/1_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/mild_gt/1_gt.png -------------------------------------------------------------------------------- /data/test/mild_gt/2_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/mild_gt/2_gt.png -------------------------------------------------------------------------------- /data/test/mild_gt/3_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/mild_gt/3_gt.png -------------------------------------------------------------------------------- /data/test/mild_gt/4_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/mild_gt/4_gt.png -------------------------------------------------------------------------------- /data/test/mild_in/1_in.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/mild_in/1_in.png -------------------------------------------------------------------------------- /data/test/mild_in/2_in.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/mild_in/2_in.png -------------------------------------------------------------------------------- /data/test/mild_in/3_in.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/mild_in/3_in.png -------------------------------------------------------------------------------- /data/test/mild_in/4_in.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/mild_in/4_in.png -------------------------------------------------------------------------------- /data/test/mine/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/mine/1.jpg -------------------------------------------------------------------------------- /data/test/mine/2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/mine/2.jpg -------------------------------------------------------------------------------- /data/test/mine/3.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/mine/3.jpg -------------------------------------------------------------------------------- /data/test/mine/4.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/mine/4.jpg -------------------------------------------------------------------------------- /data/test/moderate_gt/1_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/moderate_gt/1_gt.png -------------------------------------------------------------------------------- /data/test/moderate_gt/2_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/moderate_gt/2_gt.png -------------------------------------------------------------------------------- /data/test/moderate_gt/3_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/moderate_gt/3_gt.png -------------------------------------------------------------------------------- /data/test/moderate_gt/4_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/moderate_gt/4_gt.png -------------------------------------------------------------------------------- /data/test/moderate_in/1_in.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/moderate_in/1_in.png -------------------------------------------------------------------------------- /data/test/moderate_in/2_in.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/moderate_in/2_in.png -------------------------------------------------------------------------------- /data/test/moderate_in/3_in.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/moderate_in/3_in.png -------------------------------------------------------------------------------- /data/test/moderate_in/4_in.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/moderate_in/4_in.png -------------------------------------------------------------------------------- /data/test/severe_gt/1_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/severe_gt/1_gt.png -------------------------------------------------------------------------------- /data/test/severe_gt/2_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/severe_gt/2_gt.png -------------------------------------------------------------------------------- /data/test/severe_gt/3_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/severe_gt/3_gt.png -------------------------------------------------------------------------------- /data/test/severe_gt/4_gt.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/severe_gt/4_gt.png -------------------------------------------------------------------------------- /data/test/severe_in/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/severe_in/1.png -------------------------------------------------------------------------------- /data/test/severe_in/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/severe_in/2.png -------------------------------------------------------------------------------- /data/test/severe_in/3_in.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/severe_in/3_in.png -------------------------------------------------------------------------------- /data/test/severe_in/4_in.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/test/severe_in/4_in.png -------------------------------------------------------------------------------- /data/train/generate_train.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/train/generate_train.m -------------------------------------------------------------------------------- /data/train/noise_combination.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/train/noise_combination.m -------------------------------------------------------------------------------- /data/train/store2hdf5.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/train/store2hdf5.m -------------------------------------------------------------------------------- /data/valid/validation.h5: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/data/valid/validation.h5 -------------------------------------------------------------------------------- /dqn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /dqn/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/dqn/agent.py -------------------------------------------------------------------------------- /dqn/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/dqn/base.py -------------------------------------------------------------------------------- /dqn/environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/dqn/environment.py -------------------------------------------------------------------------------- /dqn/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/dqn/ops.py -------------------------------------------------------------------------------- /dqn/replay_memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/dqn/replay_memory.py -------------------------------------------------------------------------------- /dqn/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/dqn/utils.py -------------------------------------------------------------------------------- /imgs/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/imgs/framework.png -------------------------------------------------------------------------------- /imgs/real_world.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/imgs/real_world.png -------------------------------------------------------------------------------- /imgs/restore.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/imgs/restore.gif -------------------------------------------------------------------------------- /imgs/tensorboard.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/imgs/tensorboard.png -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/main.py -------------------------------------------------------------------------------- /models/checkpoint: -------------------------------------------------------------------------------- 1 | model_checkpoint_path: "model" 2 | -------------------------------------------------------------------------------- /models/model.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/models/model.data-00000-of-00001 -------------------------------------------------------------------------------- /models/model.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/models/model.index -------------------------------------------------------------------------------- /models/model.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/models/model.meta -------------------------------------------------------------------------------- /toolbox/checkpoint: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/checkpoint -------------------------------------------------------------------------------- /toolbox/tool01.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool01.data-00000-of-00001 -------------------------------------------------------------------------------- /toolbox/tool01.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool01.index -------------------------------------------------------------------------------- /toolbox/tool01.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool01.meta -------------------------------------------------------------------------------- /toolbox/tool02.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool02.data-00000-of-00001 -------------------------------------------------------------------------------- /toolbox/tool02.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool02.index -------------------------------------------------------------------------------- /toolbox/tool02.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool02.meta -------------------------------------------------------------------------------- /toolbox/tool03.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool03.data-00000-of-00001 -------------------------------------------------------------------------------- /toolbox/tool03.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool03.index -------------------------------------------------------------------------------- /toolbox/tool03.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool03.meta -------------------------------------------------------------------------------- /toolbox/tool04.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool04.data-00000-of-00001 -------------------------------------------------------------------------------- /toolbox/tool04.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool04.index -------------------------------------------------------------------------------- /toolbox/tool04.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool04.meta -------------------------------------------------------------------------------- /toolbox/tool05.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool05.data-00000-of-00001 -------------------------------------------------------------------------------- /toolbox/tool05.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool05.index -------------------------------------------------------------------------------- /toolbox/tool05.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool05.meta -------------------------------------------------------------------------------- /toolbox/tool06.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool06.data-00000-of-00001 -------------------------------------------------------------------------------- /toolbox/tool06.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool06.index -------------------------------------------------------------------------------- /toolbox/tool06.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool06.meta -------------------------------------------------------------------------------- /toolbox/tool07.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool07.data-00000-of-00001 -------------------------------------------------------------------------------- /toolbox/tool07.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool07.index -------------------------------------------------------------------------------- /toolbox/tool07.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool07.meta -------------------------------------------------------------------------------- /toolbox/tool08.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool08.data-00000-of-00001 -------------------------------------------------------------------------------- /toolbox/tool08.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool08.index -------------------------------------------------------------------------------- /toolbox/tool08.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool08.meta -------------------------------------------------------------------------------- /toolbox/tool09.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool09.data-00000-of-00001 -------------------------------------------------------------------------------- /toolbox/tool09.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool09.index -------------------------------------------------------------------------------- /toolbox/tool09.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool09.meta -------------------------------------------------------------------------------- /toolbox/tool10.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool10.data-00000-of-00001 -------------------------------------------------------------------------------- /toolbox/tool10.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool10.index -------------------------------------------------------------------------------- /toolbox/tool10.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool10.meta -------------------------------------------------------------------------------- /toolbox/tool11.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool11.data-00000-of-00001 -------------------------------------------------------------------------------- /toolbox/tool11.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool11.index -------------------------------------------------------------------------------- /toolbox/tool11.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool11.meta -------------------------------------------------------------------------------- /toolbox/tool12.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool12.data-00000-of-00001 -------------------------------------------------------------------------------- /toolbox/tool12.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool12.index -------------------------------------------------------------------------------- /toolbox/tool12.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yuke93/RL-Restore/HEAD/toolbox/tool12.meta --------------------------------------------------------------------------------