Giter Club home page Giter Club logo

tracklet_parser's Introduction

Tracklet Parser

license python black

A parser for tracklet labels in KITTI Raw Format 1.0 created by the Computer Vision Annotation Tool (CVAT).

Installation

Assuming that you already have a working Python environment, you can install all necessary packages with

python -m pip install .

Usage

Creating a folder with text files in KITTI format. Each text file contains the labeling information of its corresponding recording.

from tracklet_parser.tracklet import Tracklet
from tracklet_parser.tracklet_parser import TrackletParser

def main():
    tracklet_labels: str = "C:\\Foo\\tracklet_labels.xml"
    frame_list: str = "C:\\Foo\\frame_list.txt"
    output_dir: str = "C:\\Foo\\Bar"

    tracklets: List[Tracklet] = TrackletParser.parse_tracklet_xml(tracklet_labels)
    # Create n label text files in C:\Foo\Bar for n (labeled) recordings
    TrackletParser.convert_tracklets_to_kitti(tracklets, frame_list, output_dir)

if __name__ == "__main__":
    main()

tracklet_parser's People

Contributors

holtvogt avatar

Stargazers

 avatar  avatar  avatar  avatar

Watchers

 avatar  avatar

tracklet_parser's Issues

Not able to import TrackletParser

Describe the bug
Hi,

I am trying to implement tracklet_parser on Google Colab. I cloned the repository and ran the setup.py file. For the next step as shown on in https://github.com/holtvogt/tracklet_parser, I am trying to import the TrackletParser. But it is not able to import it and shows error.

To Reproduce
Steps to reproduce the behavior:

  1. Go to Google Colab Notebook.
  2. Clone the repo using !git clone https://github.com/holtvogt/tracklet_parser
  3. Set the current directory using %cd /content/tracklet_parser
  4. Run the setup.py file using !python setup.py install
  5. Import TrackletParser using from tracklet_parser import TrackletParser
  6. See error.

Expected behavior
The TrackletParser should be imported. However, I see error on importing.

Screenshots
The screenshot of the error is attached below.
image

Please let me know how to handle this.

Add argument parser for script automation

Describe the feature
Adding an argument parser to its initial functionality.

Motivation
As some might want to automate the steps to create label files, It'd be handy to be able to execute the tracklet parser via command line with arguments. E.g.

python parser.py --labels <tracklet_labels.xml> --frames <frame_list.txt> --out <output_dir>

Rotation angle in tracklet_labels.xml

When labeling my point cloud data via CVAT, I observed that the only meaningful rotation angle that changes, is the rotation angle z. This makes sense in a manner that in LiDAR coordinates, this is the yaw axis.

Maybe this query

elif pose_attribute.tag == "ry":

needs to be adjusted to the proper camera coordinate conversion (as y in camera coordinates is -z in LiDAR coordinates):

elif pose_attribute.tag == "rz":
    tracklet.rotation_y = -float(pose_attribute.text)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.