Home

Awesome

MLNet-Pytorch

Implementation of A Deep Multi-Level Network for Saliency Prediction in Pytorch

Description :

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.

For Mobile Users: https://nbviewer.jupyter.org/github/immortal3/MLNet-Pytorch/blob/master/MLNet_Pytorch.ipynb

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.

Result :

AUCCCKLSAUCIGNSSSIM
Ours0.7710.5531.1420.619-0.3801.0140.573
Original Image1234
Predicted Saliency Map1234
Ground Truth3333

References:

[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.