nuScenes-converter

Using nuScenes with detectron2

For what

Detectron2 doesn't support nuScence annotation out of the box. To use this dataset, you must transform annotation to the dictionary (see dict format here). This package does it for you. Also you can use it like 3b-box to 2d-bbox translator.

Requirements

• Python ≥ 3.6
• PyTorch ≥ 1.4
• OpenCV
• detectron2
• nuScenes devkit
• nuScenes, demo use mini version

Installation

git clone https://github.com/tdtce/nuScenes-converter.git
cd nuScenes-converter
pip install -e .

Using example

Using example see here. For import use:

from nusc_conv.conv import get_nuscenes_dicts

Plans

• Category support
• Train test support