cvhciKIT / sloth

Sloth is a tool for labeling image and video data for computer vision research.
Other
608 stars 199 forks source link

Is it possible to assign labels to shapes? #145

Open mrgloom opened 6 years ago

mrgloom commented 6 years ago

Is it possible to assign labels to shapes? For example is it possible to assign label (for example 'cat', 'dog') for polygon shapes?

mrgloom commented 6 years ago

As I understand it can be done via custom config:

Also you can see deafult config under: C:\Users\myuser\Downloads\sloth-master\sloth-master\sloth\conf\default_config.py

from PyQt4.QtGui import QPen
from PyQt4.Qt import Qt
from sloth.items import PolygonItem
from sloth.items import PolygonItemInserter

LABELS = (
    {
        'attributes': {
            'type':  'polygon',
            'class': 'animals',
            'id':    ['dog', 'cat']
        },
        'inserter': 'sloth.items.PolygonItemInserter',
        'item':     'sloth.items.PolygonItem',
        'hotkey':   'o',
        'text':     'Polygon',
    },
)

And then run sloth like:

C:\Users\myuser\Downloads\sloth-master\sloth-master>C:\Users\myuser\Anaconda2\py
thon.exe sloth/bin/sloth --config examples/animals_config.py

But one problem, after I have finished one polygon I need again select polygon button or press o key, I want to automate this part too, is it possible? Also is it possible to bind hot keys to 'cat', 'dog' classes (for example bind '1' on keyboard to 'cat' class and '2' on keyboard to 'dog' class)?