├── README.md ├── assets ├── Detection.jpg ├── ablation.jpg ├── app.jpg ├── compare.jpg ├── framework.jpg ├── frequency.jpg ├── highlight.jpg ├── illumination.jpg ├── introduction.jpg ├── levels.jpg └── realworld.jpg ├── data └── prompt_dataset.py ├── dataset ├── Infrared │ ├── 00079.png │ ├── 00102.png │ └── 010389.png └── Visible │ ├── 00079.png │ ├── 00102.png │ └── 010389.png ├── genDateset ├── ir_lowcontrast.py ├── ir_noise.py ├── ir_stripe_noise.py ├── vis_blur.py ├── vis_haze.py ├── vis_noise.py └── vis_rain.py ├── model ├── Adapter_both.py ├── Adapter_one.py └── ControlFusion.py ├── pretrained_weights └── put_them_here.txt ├── requirements.txt ├── scripts ├── losses.py └── utils.py ├── test.py ├── train.py └── transforms.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/README.md -------------------------------------------------------------------------------- /assets/Detection.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/assets/Detection.jpg -------------------------------------------------------------------------------- /assets/ablation.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/assets/ablation.jpg -------------------------------------------------------------------------------- /assets/app.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/assets/app.jpg -------------------------------------------------------------------------------- /assets/compare.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/assets/compare.jpg -------------------------------------------------------------------------------- /assets/framework.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/assets/framework.jpg -------------------------------------------------------------------------------- /assets/frequency.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/assets/frequency.jpg -------------------------------------------------------------------------------- /assets/highlight.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/assets/highlight.jpg -------------------------------------------------------------------------------- /assets/illumination.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/assets/illumination.jpg -------------------------------------------------------------------------------- /assets/introduction.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/assets/introduction.jpg -------------------------------------------------------------------------------- /assets/levels.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/assets/levels.jpg -------------------------------------------------------------------------------- /assets/realworld.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/assets/realworld.jpg -------------------------------------------------------------------------------- /data/prompt_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/data/prompt_dataset.py -------------------------------------------------------------------------------- /dataset/Infrared/00079.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/dataset/Infrared/00079.png -------------------------------------------------------------------------------- /dataset/Infrared/00102.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/dataset/Infrared/00102.png -------------------------------------------------------------------------------- /dataset/Infrared/010389.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/dataset/Infrared/010389.png -------------------------------------------------------------------------------- /dataset/Visible/00079.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/dataset/Visible/00079.png -------------------------------------------------------------------------------- /dataset/Visible/00102.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/dataset/Visible/00102.png -------------------------------------------------------------------------------- /dataset/Visible/010389.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/dataset/Visible/010389.png -------------------------------------------------------------------------------- /genDateset/ir_lowcontrast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/genDateset/ir_lowcontrast.py -------------------------------------------------------------------------------- /genDateset/ir_noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/genDateset/ir_noise.py -------------------------------------------------------------------------------- /genDateset/ir_stripe_noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/genDateset/ir_stripe_noise.py -------------------------------------------------------------------------------- /genDateset/vis_blur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/genDateset/vis_blur.py -------------------------------------------------------------------------------- /genDateset/vis_haze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/genDateset/vis_haze.py -------------------------------------------------------------------------------- /genDateset/vis_noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/genDateset/vis_noise.py -------------------------------------------------------------------------------- /genDateset/vis_rain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/genDateset/vis_rain.py -------------------------------------------------------------------------------- /model/Adapter_both.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/model/Adapter_both.py -------------------------------------------------------------------------------- /model/Adapter_one.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/model/Adapter_one.py -------------------------------------------------------------------------------- /model/ControlFusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/model/ControlFusion.py -------------------------------------------------------------------------------- /pretrained_weights/put_them_here.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/scripts/losses.py -------------------------------------------------------------------------------- /scripts/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/scripts/utils.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/train.py -------------------------------------------------------------------------------- /transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Linfeng-Tang/ControlFusion/HEAD/transforms.py --------------------------------------------------------------------------------