Skip to content

GuillaumeBalezo/SOD-python

 
 

Repository files navigation

Unconstrained Salient Object Detection

License

This is my implementation of the salient object detection method described in

Jianming Zhang, Stan Sclaroff, Zhe Lin, Xiaohui Shen, Brian Price and Radomír Mech. "Unconstrained Salient Object Detection via Proposal Subset Optimization." CVPR, 2016.

The original implementation was in Matlab and Caffe. I converted it in Python and Tensorflow.

This method aims at producing a highly compact set of detection windows for salient objects in uncontrained images, which may or may not contain salient objects. Please cite the above paper if you find this work useful.

Alt text

Prerequisites

  1. Linux
  2. Python 3
  3. Tensorflow 2

Quick Start

  1. Unzip the files to a local folder.
  2. Download the weights
  3. Donwload the MSO dataset
  4. Run demo.py.

You can also run the jupyter notebook. If you don't want to install anything, you can execute the notebook with Google Colab.

Evaluation

You can reproduce the result on the MSO dataset reported in the paper, by run benchmark_MSO.py. It will automatically download the MSO dataset and the pre-trained VGG16 model.

The results are the same as the matlab implementation.

Miscs

To change some configurations, please check get_Param.py.

There is an heuristic window refining process for small objects like in the matlab implementation. Note that this process is not included in the paper or used in the evaluation.

About

My Python and Tensorflow implementation of a Method for Unconstrained Salient Object Detection.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Jupyter Notebook 98.1%
  • Python 1.9%