qcjames53 / annotationtool

Python library allowing for 3D annotation of KITTI images in arbitrary reference planes.

AnnotationTool

AnnotationTool is a Python 3.6 library that allows for the manual 3D annotation of images in arbitrary reference planes. The program allows for the tracking of arbitrary types of objects, outputting each object's position in a format consistent with the KITTI data set when requested.

This library is actively under development. Future plans revolve around implementing this tool as a method for visualizing and training the M3D-RPN library, from Michigan State's computer vision lab.

Installation

The library currently requires Python 3.6 to help mitigate potential issues with integrating M3D-RPN in the future.

Required installations:

pip install numpy
pip install pygame

Usage

AnnotationTool can be run via python main.py. It is preconfigured to load a set amount of low-filesize .jpg images from the /images_2 directory. Available button controls are displayed. Holding CTRL will increase the output of many buttons.