├── LICENSE ├── README.md ├── VOC2012_224_train_png.txt ├── data ├── VOC2012_224_train_png.txt ├── __init__.py ├── demoire_dataset.py ├── denoising_dataset.py ├── image_folder.py ├── intrinsic_dataset.py ├── real_test.txt ├── sirs_dataset.py ├── torchdata.py └── transforms.py ├── engine.py ├── figures ├── figure_eval_comparation.png ├── figure_eval_demoire.png ├── figure_eval_denoising.png ├── figure_eval_intrinsic_decomp.png ├── figure_test_comparation.png └── figure_ytmt_networks_final.png ├── models ├── __init__.py ├── arch │ ├── __init__.py │ ├── dncnn.py │ ├── unet.py │ ├── ytmt_networks.py │ └── ~ │ │ ├── YTMT_US │ │ ├── YTMT_US.pdf │ │ └── tmp.pth ├── base_model.py ├── losses.py ├── losses_opt.py ├── networks.py ├── opt_ytmt_model_sirs.py ├── twostage_ytmt_model.py ├── twostage_ytmt_model_opt.py ├── vgg.py ├── ytmt_model_demoire.py ├── ytmt_model_intrinsic_decomp.py └── ytmt_model_sirs.py ├── options ├── __init__.py ├── base_option.py └── net_options │ ├── __init__.py │ ├── base_options.py │ └── train_options.py ├── test_demoire.py ├── test_denoising.py ├── test_intrinsic.py ├── test_sirs.py ├── tools ├── TensorboardWriter.py ├── __init__.py ├── math_utils.py ├── model_utils.py ├── mutils.py └── saver.py ├── train_demoire.py ├── train_denoising.py ├── train_intrinsic.py ├── train_sirs.py ├── train_sirs_twostage.py └── util ├── __init__.py ├── html.py ├── index.py ├── net_utils.py ├── schedulers.py ├── util.py └── visualizer.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/README.md -------------------------------------------------------------------------------- /VOC2012_224_train_png.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/VOC2012_224_train_png.txt -------------------------------------------------------------------------------- /data/VOC2012_224_train_png.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/data/VOC2012_224_train_png.txt -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/demoire_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/data/demoire_dataset.py -------------------------------------------------------------------------------- /data/denoising_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/data/denoising_dataset.py -------------------------------------------------------------------------------- /data/image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/data/image_folder.py -------------------------------------------------------------------------------- /data/intrinsic_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/data/intrinsic_dataset.py -------------------------------------------------------------------------------- /data/real_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/data/real_test.txt -------------------------------------------------------------------------------- /data/sirs_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/data/sirs_dataset.py -------------------------------------------------------------------------------- /data/torchdata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/data/torchdata.py -------------------------------------------------------------------------------- /data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/data/transforms.py -------------------------------------------------------------------------------- /engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/engine.py -------------------------------------------------------------------------------- /figures/figure_eval_comparation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/figures/figure_eval_comparation.png -------------------------------------------------------------------------------- /figures/figure_eval_demoire.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/figures/figure_eval_demoire.png -------------------------------------------------------------------------------- /figures/figure_eval_denoising.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/figures/figure_eval_denoising.png -------------------------------------------------------------------------------- /figures/figure_eval_intrinsic_decomp.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/figures/figure_eval_intrinsic_decomp.png -------------------------------------------------------------------------------- /figures/figure_test_comparation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/figures/figure_test_comparation.png -------------------------------------------------------------------------------- /figures/figure_ytmt_networks_final.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/figures/figure_ytmt_networks_final.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/arch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/arch/__init__.py -------------------------------------------------------------------------------- /models/arch/dncnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/arch/dncnn.py -------------------------------------------------------------------------------- /models/arch/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/arch/unet.py -------------------------------------------------------------------------------- /models/arch/ytmt_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/arch/ytmt_networks.py -------------------------------------------------------------------------------- /models/arch/~/YTMT_US: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/arch/~/YTMT_US -------------------------------------------------------------------------------- /models/arch/~/YTMT_US.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/arch/~/YTMT_US.pdf -------------------------------------------------------------------------------- /models/arch/~/tmp.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/arch/~/tmp.pth -------------------------------------------------------------------------------- /models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/base_model.py -------------------------------------------------------------------------------- /models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/losses.py -------------------------------------------------------------------------------- /models/losses_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/losses_opt.py -------------------------------------------------------------------------------- /models/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/networks.py -------------------------------------------------------------------------------- /models/opt_ytmt_model_sirs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/opt_ytmt_model_sirs.py -------------------------------------------------------------------------------- /models/twostage_ytmt_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/twostage_ytmt_model.py -------------------------------------------------------------------------------- /models/twostage_ytmt_model_opt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/twostage_ytmt_model_opt.py -------------------------------------------------------------------------------- /models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/vgg.py -------------------------------------------------------------------------------- /models/ytmt_model_demoire.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/ytmt_model_demoire.py -------------------------------------------------------------------------------- /models/ytmt_model_intrinsic_decomp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/ytmt_model_intrinsic_decomp.py -------------------------------------------------------------------------------- /models/ytmt_model_sirs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/models/ytmt_model_sirs.py -------------------------------------------------------------------------------- /options/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /options/base_option.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/options/base_option.py -------------------------------------------------------------------------------- /options/net_options/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /options/net_options/base_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/options/net_options/base_options.py -------------------------------------------------------------------------------- /options/net_options/train_options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/options/net_options/train_options.py -------------------------------------------------------------------------------- /test_demoire.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/test_demoire.py -------------------------------------------------------------------------------- /test_denoising.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/test_denoising.py -------------------------------------------------------------------------------- /test_intrinsic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/test_intrinsic.py -------------------------------------------------------------------------------- /test_sirs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/test_sirs.py -------------------------------------------------------------------------------- /tools/TensorboardWriter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/tools/TensorboardWriter.py -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/tools/__init__.py -------------------------------------------------------------------------------- /tools/math_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/tools/math_utils.py -------------------------------------------------------------------------------- /tools/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/tools/model_utils.py -------------------------------------------------------------------------------- /tools/mutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/tools/mutils.py -------------------------------------------------------------------------------- /tools/saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/tools/saver.py -------------------------------------------------------------------------------- /train_demoire.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/train_demoire.py -------------------------------------------------------------------------------- /train_denoising.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/train_denoising.py -------------------------------------------------------------------------------- /train_intrinsic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/train_intrinsic.py -------------------------------------------------------------------------------- /train_sirs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/train_sirs.py -------------------------------------------------------------------------------- /train_sirs_twostage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/train_sirs_twostage.py -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/util/__init__.py -------------------------------------------------------------------------------- /util/html.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/util/html.py -------------------------------------------------------------------------------- /util/index.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/util/index.py -------------------------------------------------------------------------------- /util/net_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/util/net_utils.py -------------------------------------------------------------------------------- /util/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/util/schedulers.py -------------------------------------------------------------------------------- /util/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/util/util.py -------------------------------------------------------------------------------- /util/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mingcv/YTMT-Strategy/HEAD/util/visualizer.py --------------------------------------------------------------------------------