# This is official Pytorch implementation of "ControlFusion: A Controllable Image Fusion Framework with Language-Vision Degradation Prompts"

## Fast Testing

1. **Run the following code for faster testing:**

   ```
   Downloading the [pre-trained model]() and placing them in **./pretrained** (Due to policy limitations we are unable to provide download links. We will update the links as they become available.)
   python test_from_dataset.py
   ```
2. **Run the following code for Stage I training:**

   ```
   python train_adapter.py
   ```
3. **Run the following code for Stage II training:**

   ```
   python train_fusion.py
   ```
