pyqt5-video-labeling

This tool is available to label in video (only bounding box so far).

Dependencies

Develop the application with pipenv + python 3.6.5 as follow, or you can simply install the pyqt5 and opencv

better-exceptions==0.2.1
opencv-python==3.4.3.18
  - numpy [required: >=1.11.3, installed: 1.15.1]
pandas==0.23.4
  - numpy [required: >=1.9.0, installed: 1.15.1]
  - python-dateutil [required: >=2.5.0, installed: 2.7.3]
    - six [required: >=1.5, installed: 1.11.0]
  - pytz [required: >=2011k, installed: 2018.5]
pylint==2.1.1
  - astroid [required: >=2.0.0, installed: 2.0.4]
    - lazy-object-proxy [required: Any, installed: 1.3.1]
    - six [required: Any, installed: 1.11.0]
    - typed-ast [required: Any, installed: 1.1.0]
    - wrapt [required: Any, installed: 1.10.11]
  - isort [required: >=4.2.5, installed: 4.3.4]
  - mccabe [required: Any, installed: 0.6.1]
PyQt5==5.11.2
  - PyQt5-sip [required: <4.20,>=4.19.11, installed: 4.19.12]
PyYAML==3.13

Install

Require pipenv to install dependecy

$ git clone https://github.com/afunTW/pyqt5-video-labeling
$ cd pyqt5-video-labeling
$ pipenv sync

Usage

$ python3 main.py -h
usage: main.py [-h] [-v VIDEO] [-c CONFIG] [-o OUTPUT]

optional arguments:
  -h, --help            show this help message and exit
  -v VIDEO, --video VIDEO
  -c CONFIG, --config CONFIG
  -o OUTPUT, --output OUTPUT

Functionality

event	target	function
RIGHT_CLICK	a record on the frame	DELETE
DOUBLE CLICK	a record on the right table widget	skip to frame
KEY_SPACE or KEY_P	KEYBOARD	play video
KEY_RIGHT or KEY_D	KEYBOARD	skip to next second frame
KEY_LEFT or KEY_A	KEYBOARD	skip to previous second frame

More configuration can be modified in config.yaml, including

• color, thickness, line_style in drawing mode
• color, thickness, line_style in selecting mode
• color, thickness in label
• limit number of labels in single frame