A parser for tracklet labels in KITTI Raw Format 1.0 created by the Computer Vision Annotation Tool (CVAT).
Install the package directly using pip from PyPI:
pip install tracklet-parserTo use the tracklet-parser, you need a tracklet_labels.xml file and, optionally, a frame_list.txt file that maps frame indices to point cloud file names. The parser generates one KITTI format label file per frame, storing them in the specified output directory.
Creating KITTI format label files from tracklet data is straightforward with the tracklet-parser. Each output file corresponds to a single frame and contains labeling information for all detected objects in that frame (e.g., Car, Pedestrian, etc.).
Here is an example script demonstrating how to parse a tracklet_labels.xml file and convert the tracklet data into KITTI format label files:
from tracklet_parser.tracklet import Tracklet
from tracklet_parser.tracklet_parser import TrackletParser
def main():
tracklet_labels: str = "path/to/tracklet_labels.xml"
frame_list: str = "path/to/frame_list.txt"
output_dir: str = "path/to/output_dir"
tracklets: List[Tracklet] = TrackletParser.parse_tracklet_xml(tracklet_labels)
# Create n label text files in path/to/output_dir for n (labeled) recordings in KITTI format
TrackletParser.convert_tracklets_to_kitti(tracklets, frame_list, output_dir)
if __name__ == "__main__":
main()To run the tests, use the following command:
python -m unittest discover testsContributions are welcome! To contribute:
- Fork the repository.
- Create a new branch for your feature or bugfix.
- Submit a pull request with a clear description of your changes.
Please ensure your code adheres to the Black coding style.