zbxzc35 / MST-python

A python re-implementation of Real-Time Salient Object Detection with a Minimum Spanning Tree, CVPR 2016

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Real-Time Salient Object Detection with a Minimum Spanning Tree

salmap

This is a python re-implementation of a paper published in CVPR 2016, which can be downloaded in here

Environment you need is: Python 3, cv2 and numpy. run demo.py you will get the salmap (like the right figure). It takes about 20 seconds in my personal laptop.

There is a c++ re-implementation which is much much faster than me in here . My whole algorithm is based on this and the variable name is the same as him.

Due to the subtle differences of the sort algorithm, few pixel value of my result is slightly different.

By the way, if you use the code, please cite the following reference:

@inproceedings{Tu2016Real,
  title={Real-Time Salient Object Detection with a Minimum Spanning Tree},
  author={Tu, Wei Chih and He, Shengfeng and Yang, Qingxiong and Chien, Shao Yi},
  booktitle={IEEE Conference on Computer Vision & Pattern Recognition},
  year={2016},
}

About

A python re-implementation of Real-Time Salient Object Detection with a Minimum Spanning Tree, CVPR 2016

License:MIT License


Languages

Language:Python 100.0%