mirror of
https://github.com/wassname/sloth.git
synced 2026-07-13 02:21:08 +08:00
Mike Gerber
3358d6d579
Add the functionality to add labels of polygon shape. That means: - Fix the non-working PolygonItemInserter class - Add the missing PolygonItem class - Add the "polygon" class to the default configuration
124 lines
4.8 KiB
Python
124 lines
4.8 KiB
Python
# This is sloth's default configuration.
|
|
#
|
|
# The configuration file is a simple python module with module-level
|
|
# variables. This module contains the default values for sloth's
|
|
# configuration variables.
|
|
#
|
|
# In all cases in the configuration where a python callable (such as a
|
|
# function, class constructor, etc.) is expected, it is equally possible
|
|
# to specify a module path (as string) pointing to such a python callable.
|
|
# It will then be automatically imported.
|
|
|
|
# LABELS
|
|
#
|
|
# List/tuple of dictionaries that defines the label classes
|
|
# that are handled by sloth. For each label, there should
|
|
# be one dictionary that contains the following keys:
|
|
#
|
|
# - 'item' : Visualization item for this label. This can be
|
|
# any python callable or a module path string
|
|
# implementing the visualization item interface.
|
|
#
|
|
# - 'inserter' : (optional) Item inserter for this label.
|
|
# If the user selects to insert a new label of this class
|
|
# the inserter is responsible to actually
|
|
# capture the users mouse actions and insert
|
|
# a new label into the annotation model.
|
|
#
|
|
# - 'hotkey' : (optional) A keyboard shortcut starting
|
|
# the insertion of a new label of this class.
|
|
#
|
|
# - 'attributes' : (optional) A dictionary that defines the
|
|
# keys and possible values of this label
|
|
# class.
|
|
#
|
|
# - 'text' : (optional) A label for the item's GUI button.
|
|
LABELS = (
|
|
{
|
|
'attributes': {
|
|
'class': 'Face',
|
|
},
|
|
'inserter': 'sloth.items.RectItemInserter',
|
|
'item': 'sloth.items.RectItem',
|
|
'hotkey': 'f',
|
|
'text': 'Face',
|
|
},
|
|
{
|
|
'attributes': {
|
|
'class': 'rect',
|
|
},
|
|
'inserter': 'sloth.items.RectItemInserter',
|
|
'item': 'sloth.items.RectItem',
|
|
'hotkey': 'r',
|
|
'text': 'Rectangle',
|
|
},
|
|
{
|
|
'attributes': {
|
|
'class': 'point',
|
|
},
|
|
'inserter': 'sloth.items.PointItemInserter',
|
|
'item': 'sloth.items.PointItem',
|
|
'hotkey': 'p',
|
|
'text': 'Point',
|
|
},
|
|
{
|
|
'attributes': {
|
|
'class': 'polygon',
|
|
},
|
|
'inserter': 'sloth.items.PolygonItemInserter',
|
|
'item': 'sloth.items.PolygonItem',
|
|
'hotkey': 'o',
|
|
'text': 'Polygon',
|
|
},
|
|
)
|
|
|
|
# HOTKEYS
|
|
#
|
|
# Defines the keyboard shortcuts. Each hotkey is defined by a tuple
|
|
# with at least 2 entries, where the first entry is the hotkey (sequence),
|
|
# and the second entry is the function that is called. The function
|
|
# should expect a single parameter, the labeltool object. The optional
|
|
# third entry -- if present -- is expected to be a string describing the
|
|
# action.
|
|
HOTKEYS = (
|
|
('Space', [lambda lt: lt.currentImage().confirmAll(),
|
|
lambda lt: lt.currentImage().setUnlabeled(False),
|
|
lambda lt: lt.gotoNext()
|
|
], 'Mark image as labeled/confirmed and go to next'),
|
|
('Backspace', lambda lt: lt.gotoPrevious(), 'Previous image/frame'),
|
|
('PgDown', lambda lt: lt.gotoNext(), 'Next image/frame'),
|
|
('PgUp', lambda lt: lt.gotoPrevious(), 'Previous image/frame'),
|
|
('Tab', lambda lt: lt.selectNextAnnotation(), 'Select next annotation'),
|
|
('Shift+Tab', lambda lt: lt.selectPreviousAnnotation(), 'Select previous annotation'),
|
|
('Ctrl+f', lambda lt: lt.view().fitInView(), 'Fit current image/frame into window'),
|
|
('Del', lambda lt: lt.deleteSelectedAnnotations(), 'Delete selected annotations'),
|
|
('ESC', lambda lt: lt.exitInsertMode(), 'Exit insert mode'),
|
|
('Shift+l', lambda lt: lt.currentImage().setUnlabeled(False), 'Mark current image as labeled'),
|
|
('Shift+c', lambda lt: lt.currentImage().confirmAll(), 'Mark all annotations in image as confirmed'),
|
|
)
|
|
|
|
# CONTAINERS
|
|
#
|
|
# A list/tuple of two-tuples defining the mapping between filename pattern and
|
|
# annotation container classes. The filename pattern can contain wildcards
|
|
# such as * and ?. The corresponding container is expected to either a python
|
|
# class implementing the sloth container interface, or a module path pointing
|
|
# to such a class.
|
|
CONTAINERS = (
|
|
('*.json', 'sloth.annotations.container.JsonContainer'),
|
|
('*.msgpack', 'sloth.annotations.container.MsgpackContainer'),
|
|
('*.yaml', 'sloth.annotations.container.YamlContainer'),
|
|
('*.pickle', 'sloth.annotations.container.PickleContainer'),
|
|
('*.sloth-init', 'sloth.annotations.container.FileNameListContainer'),
|
|
)
|
|
|
|
# PLUGINS
|
|
#
|
|
# A list/tuple of classes implementing the sloth plugin interface. The
|
|
# classes can either be given directly or their module path be specified
|
|
# as string.
|
|
PLUGINS = (
|
|
)
|
|
|
|
|