Implementation of A Deep Multi-Level Network for Saliency Prediction in Pytorch
As Human, We only focus on certain part of Image which is called salient region. This Project predict Saliency map of given Image which is useful in many areas like Robotics, Image-aware editing, caption generation, fast-response systems.
Notebook is directly runnable to Google colab.
Note : Due to Memory Limit, some layers were frozen and output saliency maps size was reduced by half during Training.
|
AUC |
CC |
KL |
SAUC |
IG |
NSS |
SIM |
| Ours |
0.771 |
0.553 |
1.142 |
0.619 |
-0.380 |
1.014 |
0.573 |
|
|
|
|
|
| Original Image |
 |
 |
 |
 |
| Predicted Saliency Map |
 |
 |
 |
 |
| Ground Truth |
 |
 |
 |
 |
[1] Cornia, Marcella, et al. "A deep multi-level network for saliency prediction." Pattern Recognition (ICPR), 2016 23rd International Conference on. IEEE, 2016.
[2] Jiang, Ming, et al. "Salicon: Saliency in context." Proceedings of the IEEE conference on computer vision and pattern recognition. 2015.