Skip to content

Dominoer/bmvc2020_image_harmonization

master
Switch branches/tags

Name already in use

A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Image Harmonization with Attention-based Deep Feature Modulation

This code provides a pytorch implementation of "Image Harmonization with Attention-based Deep Feature Modulation".

Project Page

Requirements

  • Python 3.7
  • PyTorch tested on 1.4.0
  • PIL
  • skimage
  • tqdm
  • numpy

Dataset

Download the iHarmony4 dataset from this link.

We resized all training images of HAdobe5k subset with a max size of 1024. Original images of HAdobe5k are significantly large, which slow down the data loading process. In order to speed up this process, we resize all training images of HAdobe5k dataset.

To resize the training images of HAdobe5k, specify the paths in the following script and run the script.

python resize.py

Training

Specify paths in the train.sh script before runing the script.

bash train.sh

Testing

Here we provide a better model compared to the model used in our paper. Although the performance on the 'ALL' dataset is the same as in the paper, this model outperforms previous methods on all sub-datasets.

To download the model, run the following script.

bash download_model.sh

To test the performances, run the following script.

bash test.sh

Citation

If you find the code useful in your research, please consider citing our paper:

@InProceedings{Hao2020bmcv,
author       = "Guoqing Hao and Satoshi Iizuka and Kazuhiro Fukui",
title        = "Image Harmonization with Attention-based Deep Feature Modulation",
booktitle    = "The British Machine Vision Conference (BMCV)",
year         = "2020",
}

About

[BMVC2020] Image Harmonization with Attention-based Deep Feature Modulation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published