How to compute MSE in each foreground ratio range?

SamsungLabs / image_harmonization

[WACV2021] Foreground-aware Semantic Representations for Image Harmonization https://arxiv.org/abs/2006.00809

Mozilla Public License 2.0

268 stars 41 forks source link

How to compute MSE in each foreground ratio range? #3

Closed zhangboshen closed 3 years ago

zhangboshen commented 3 years ago

Hi, congregation，this is a great work!

Here I have a question about how to compute MSE (fMSE) in different foreground ratio range (0%-5%, 5%-15%, etc) , just like in your paper?

polyhex-sqr commented 3 years ago

Hi there, Sorry for such a late answer, hope it helps.

There's a script for this kind of evaluation. You can run the following: python3 scripts/evaluate_model_fg_ratios.py improved_dih256 /hdd0/harmonization_exps/fixed256/improved_dih/checkpoints/last_checkpoint.pth --resize-strategy Fixed256