vanttec / openlabeling

Label images and video for Computer Vision applications

OpenLabeling: open-source image and video labeler

Image labeling in multiple annotation formats:

• YOLO darknet

Table of contents

• Prerequisites
• Run project
• GUI usage
• Authors

Prerequisites

You need to install:

• [Python3]
• [OpenCV] version >= 3.0
• numpy, tqdm and lxml:

Running Requirements.txt:

python3 -mpip install -U pip
python3 -mpip install -U -r requirements.txt

Run project

Step by step:

1. Open the main/ directory

2. Insert the input images and videos in the folder input/

3. Insert the classes in the file class_list.txt (one class name per line)

4. Run the code:

5. You can find the annotations in the folder output/

   python3 main.py
   
   
   python3 main.py [-h] [-i] [-o] [-t] [--tracker TRACKER_TYPE] [-n N_FRAMES]
   
   optional arguments:
    -h, --help                Show this help message and exit
    -i, --input               Path to images and videos input folder | Default: input/
    -o, --output              Path to output folder (if using the PASCAL VOC format it's important to set this path correctly) | Default: output/
    -t, --thickness           Bounding box and cross line thickness (int) | Default: -t 1
    --tracker tracker_type    tracker_type being used: ['CSRT', 'KCF','MOSSE', 'MIL', 'BOOSTING', 'MEDIANFLOW', 'TLD', 'GOTURN', 'DASIAMRPN']
    -n N_FRAMES               number of frames to track object for
   

GUI usage

Keyboard, press:

Key	Description
a/d	previous/next image
s/w	previous/next class
e	edges
h	help
q	quit

Video:

Key	Description
p	predict the next frames' labels

Mouse:

• Use two separate left clicks to do each bounding box
• Right-click -> quick delete!
• Use the middle mouse to zoom in and out
• Use double click to select a bounding box

Label Custom Dataset

Modify class names

OpenLabeling/main/class_list.txt

insert dataset on

OpenLabeling/main/input/

Get output labeled data from

OpenLabeling/main/output/YOLO_darknet

Authors

• João Cartucho - Please give me your feedback: [email protected]

  Feel free to contribute