Selective Search in Python

# Selective Search

English | 简体中文

This is a complete implementation of selective search in Python. I thoroughly read the related papers [1][2][3] and the author’s MATLAB implementation. Compared with other implementations, my method is authentically shows the idea of the original paper. Moreover, this method has clear logic and rich annotations, which is very suitable for teaching purposes, allowing people who have just entered the CV field to understand the basic principles of selective search and exercise code reading ability.

## Installation

Installing from PyPI is recommended :

$pip install selective-search  It is also possible to install the latest version from Github source: $ git clone https://github.com/ChenjieXu/selective_search.git
$cd selective_search$ python setup.py install


Install from Anaconda:

conda install -c chenjiexu selective_search


## Quick Start

import skimage.io
from selective_search import selective_search

# Load image as NumPy array from image files

# Run selective search using single mode
boxes = selective_search(image, mode='single', random_sort=False)


For detailed examples, refer this part of the repository.

## Parameters

### Mode

Three modes correspond to various combinations of diversification strategies. The appoach to combine different diversification strategies, say, color spaces, similarity measures, starting regions is listed in the following table[1].

Mode Color Spaces Similarity Measures Starting Regions (k) Number of Combinations
single HSV CTSF 100 1
fast HSV, Lab CTSF, TSF 50, 100 8
quality HSV, Lab, rgI, H, I CTSF, TSF, F, S 50, 100, 150, 300 80
• Color Space [Source Code]
Initial oversegmentation algorithm and our subsequent grouping algorithm are performed in this colour space.

• Similarity Measure [Source Code]
'CTSF' means the similarity measure is aggregate of color similarity, texture similarity, size similarity, and fill similarity.

• Starting Region [Source Code]
A parameter of initial grouping algorithm[2], which yields high quality starting locations efficiently. A larger k causes a preference for larger components of initial strating regions.

### Random Sort

If random_sort set to True, function will carry out pseudo random sorting. It only alters sequences of bounding boxes, instead of locations, which prevents heavily emphasis on large regions as combing proposals from up to 80 different strategies[1]. This only has a significant impact when selecting a subset of region proposals with high rankings, as in RCNN.

## Project details

Uploaded source
Uploaded py3