├── README.md ├── config.yml ├── examples └── DucoNet.png ├── iharm ├── data │ ├── base.py │ ├── compose.py │ ├── hdataset.py │ └── transforms.py ├── engine │ ├── event_loop.py │ ├── optimizer.py │ └── simple_trainer.py ├── inference │ ├── evaluation.py │ ├── metrics.py │ ├── predictor.py │ ├── transforms.py │ └── utils.py ├── mconfigs │ ├── __init__.py │ ├── backboned.py │ └── base.py ├── model │ ├── backboned │ │ ├── __init__.py │ │ ├── deeplab.py │ │ ├── hrnet.py │ │ └── ih_model.py │ ├── base │ │ ├── Control_encoder.py │ │ ├── DucoNet_model.py │ │ ├── __init__.py │ │ ├── dih_model.py │ │ ├── iseunet_v1.py │ │ └── ssam_model.py │ ├── initializer.py │ ├── losses.py │ ├── metrics.py │ ├── modeling │ │ ├── basic_blocks.py │ │ ├── conv_autoencoder.py │ │ ├── deeplab_v3.py │ │ ├── hrnet_ocr.py │ │ ├── ocr.py │ │ ├── resnet.py │ │ ├── resnetv1b.py │ │ ├── styleganv2.py │ │ ├── unet.py │ │ └── unet_v1.py │ ├── modifiers.py │ ├── ops.py │ └── syncbn │ │ ├── LICENSE │ │ ├── README.md │ │ ├── __init__.py │ │ └── modules │ │ ├── __init__.py │ │ ├── functional │ │ ├── __init__.py │ │ ├── _csrc.py │ │ ├── csrc │ │ │ ├── bn.h │ │ │ ├── cuda │ │ │ │ ├── bn_cuda.cu │ │ │ │ ├── common.h │ │ │ │ └── ext_lib.h │ │ │ └── ext_lib.cpp │ │ └── syncbn.py │ │ └── nn │ │ ├── __init__.py │ │ └── syncbn.py └── utils │ ├── exp.py │ ├── log.py │ └── misc.py ├── models ├── DucoNet_1024.py ├── DucoNet_256.py └── improved_ssam.py ├── requirements.txt ├── scripts ├── evaluate_model.py ├── evaluate_model_fg_ratios.py └── predict_for_dir.py ├── test.sh ├── train.py └── train.sh /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/README.md -------------------------------------------------------------------------------- /config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/config.yml -------------------------------------------------------------------------------- /examples/DucoNet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/examples/DucoNet.png -------------------------------------------------------------------------------- /iharm/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/data/base.py -------------------------------------------------------------------------------- /iharm/data/compose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/data/compose.py -------------------------------------------------------------------------------- /iharm/data/hdataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/data/hdataset.py -------------------------------------------------------------------------------- /iharm/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/data/transforms.py -------------------------------------------------------------------------------- /iharm/engine/event_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/engine/event_loop.py -------------------------------------------------------------------------------- /iharm/engine/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/engine/optimizer.py -------------------------------------------------------------------------------- /iharm/engine/simple_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/engine/simple_trainer.py -------------------------------------------------------------------------------- /iharm/inference/evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/inference/evaluation.py -------------------------------------------------------------------------------- /iharm/inference/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/inference/metrics.py -------------------------------------------------------------------------------- /iharm/inference/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/inference/predictor.py -------------------------------------------------------------------------------- /iharm/inference/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/inference/transforms.py -------------------------------------------------------------------------------- /iharm/inference/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/inference/utils.py -------------------------------------------------------------------------------- /iharm/mconfigs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/mconfigs/__init__.py -------------------------------------------------------------------------------- /iharm/mconfigs/backboned.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/mconfigs/backboned.py -------------------------------------------------------------------------------- /iharm/mconfigs/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/mconfigs/base.py -------------------------------------------------------------------------------- /iharm/model/backboned/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/backboned/__init__.py -------------------------------------------------------------------------------- /iharm/model/backboned/deeplab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/backboned/deeplab.py -------------------------------------------------------------------------------- /iharm/model/backboned/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/backboned/hrnet.py -------------------------------------------------------------------------------- /iharm/model/backboned/ih_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/backboned/ih_model.py -------------------------------------------------------------------------------- /iharm/model/base/Control_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/base/Control_encoder.py -------------------------------------------------------------------------------- /iharm/model/base/DucoNet_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/base/DucoNet_model.py -------------------------------------------------------------------------------- /iharm/model/base/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/base/__init__.py -------------------------------------------------------------------------------- /iharm/model/base/dih_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/base/dih_model.py -------------------------------------------------------------------------------- /iharm/model/base/iseunet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/base/iseunet_v1.py -------------------------------------------------------------------------------- /iharm/model/base/ssam_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/base/ssam_model.py -------------------------------------------------------------------------------- /iharm/model/initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/initializer.py -------------------------------------------------------------------------------- /iharm/model/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/losses.py -------------------------------------------------------------------------------- /iharm/model/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/metrics.py -------------------------------------------------------------------------------- /iharm/model/modeling/basic_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/modeling/basic_blocks.py -------------------------------------------------------------------------------- /iharm/model/modeling/conv_autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/modeling/conv_autoencoder.py -------------------------------------------------------------------------------- /iharm/model/modeling/deeplab_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/modeling/deeplab_v3.py -------------------------------------------------------------------------------- /iharm/model/modeling/hrnet_ocr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/modeling/hrnet_ocr.py -------------------------------------------------------------------------------- /iharm/model/modeling/ocr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/modeling/ocr.py -------------------------------------------------------------------------------- /iharm/model/modeling/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/modeling/resnet.py -------------------------------------------------------------------------------- /iharm/model/modeling/resnetv1b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/modeling/resnetv1b.py -------------------------------------------------------------------------------- /iharm/model/modeling/styleganv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/modeling/styleganv2.py -------------------------------------------------------------------------------- /iharm/model/modeling/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/modeling/unet.py -------------------------------------------------------------------------------- /iharm/model/modeling/unet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/modeling/unet_v1.py -------------------------------------------------------------------------------- /iharm/model/modifiers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/modifiers.py -------------------------------------------------------------------------------- /iharm/model/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/ops.py -------------------------------------------------------------------------------- /iharm/model/syncbn/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/syncbn/LICENSE -------------------------------------------------------------------------------- /iharm/model/syncbn/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/syncbn/README.md -------------------------------------------------------------------------------- /iharm/model/syncbn/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/functional/__init__.py: -------------------------------------------------------------------------------- 1 | from .syncbn import batchnorm2d_sync 2 | -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/functional/_csrc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/syncbn/modules/functional/_csrc.py -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/functional/csrc/bn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/syncbn/modules/functional/csrc/bn.h -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/functional/csrc/cuda/bn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/syncbn/modules/functional/csrc/cuda/bn_cuda.cu -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/functional/csrc/cuda/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/syncbn/modules/functional/csrc/cuda/common.h -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/functional/csrc/cuda/ext_lib.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/syncbn/modules/functional/csrc/cuda/ext_lib.h -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/functional/csrc/ext_lib.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/syncbn/modules/functional/csrc/ext_lib.cpp -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/functional/syncbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/syncbn/modules/functional/syncbn.py -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/nn/__init__.py: -------------------------------------------------------------------------------- 1 | from .syncbn import * 2 | -------------------------------------------------------------------------------- /iharm/model/syncbn/modules/nn/syncbn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/model/syncbn/modules/nn/syncbn.py -------------------------------------------------------------------------------- /iharm/utils/exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/utils/exp.py -------------------------------------------------------------------------------- /iharm/utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/utils/log.py -------------------------------------------------------------------------------- /iharm/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/iharm/utils/misc.py -------------------------------------------------------------------------------- /models/DucoNet_1024.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/models/DucoNet_1024.py -------------------------------------------------------------------------------- /models/DucoNet_256.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/models/DucoNet_256.py -------------------------------------------------------------------------------- /models/improved_ssam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/models/improved_ssam.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/evaluate_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/scripts/evaluate_model.py -------------------------------------------------------------------------------- /scripts/evaluate_model_fg_ratios.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/scripts/evaluate_model_fg_ratios.py -------------------------------------------------------------------------------- /scripts/predict_for_dir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/scripts/predict_for_dir.py -------------------------------------------------------------------------------- /test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/test.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/bcmi/DucoNet-Image-Harmonization/HEAD/train.sh --------------------------------------------------------------------------------