Skip to content

Algorithm Selection platform for Object Detection

Notifications You must be signed in to change notification settings

alinoleumm/assv

Repository files navigation

Algorithm Selection Framework

Clone the repository:

$ git clone https://github.com/alinoleumm/assv.git

Install the requirements:

$ pip install -r requirements.txt

Download pre-trained algorithm selection stage I model, algorithm selection stage II model and object histograms.

Indicate pathes to downloaded files here:

cat_dir = '/path-to-object-histograms' 
wclf = joblib.load('/path-to-pretrained-as-I-model') 
iclf = joblib.load('/path-to-pretrained-as-II-model') 

Run main.py:

$ python main.py

This is a short demo of a system.

demo