Unexpected harmonization results

[galih112]

Hi,
I am testing both PCTNet and CDTNet for harmonization.
I am getting bad results when using my own images as well as when I use images provided by you.
specifically I tried libcom/tests/harmony_score_prediction/composite/ameland-5651866_harm.jpg
and its corresponding mask. (the images you are showing in the docs are not available at this repo)
any idea to what could be the problem? I am following the example provided in the docs

#Use CDTNet

CDTNet = ImageHarmonizationModel(device=0, model_type='CDTNet')
comp_img1 = image_path
comp_mask1 = mask_pth
CDT_result1 = CDTNet(comp_img1, comp_mask1)

#Use PCTNet
PCTNet = ImageHarmonizationModel(device=0, model_type='PCTNet')
comp_img1 = image_path
comp_mask1 = mask_pth
PCT_result1 = PCTNet(comp_img1, comp_mask1)

Thanks

[YujieOuO]

hi,
we provide some harmonization case in the /tests/source/composite folder.

the result of CDTNet:

For this case, the foreground color changes are not obvious due to the dark background. CDTNet is a model for global image color adjustment, which could not handle this case very well.