Folder Schema

The dataset for Modelpack is arranges as follows:

root
  - images
       - train
          - imagename1.jpg
          - imagename2.jpg
            ...
        - val
          - imagename4.jpg
          - imagename5.jpg
            ...
     
  - labels
      - train
          - imagename1.json
          - imagename2.json
            ...
      - val
          - imagename4.json
          - imagename5.json
            ...

   - dataset.yaml 

Images Files

Image files in .jpg, .jpeg or .png are acceptable

Labels File Format

File structure:

{

    "dimension":[x,y],               // x,y : int

    "image": string,                 // image filename to which this annotations belong  

    annotations as described below:

2D Boxes

    "boxes":[

        {

            "box":[x,y,w,h],           //x,y,w,h: float  - values are normalized to image width and height

            "class": string,            // label class name

            "score": float             //  value from 0 to 1 in float

        }

    ]

Segmentation Polygons

    "segment":[

      [           //Required for legacy support

        {

            "class": string ,

            "polygon":[

                  [x1,y 1],[x2, y2], ... ,[xn, yn],     // float - Vertex coordinates of polygon, normalized normalized to image with and height  

                  [x1, y1],[x2, y2], ... ,[xm, ym],    / one segmented object can have multiple polygons

            ]

        }

    ]

]

Segmentation Mask

"segmask":[

        {

            "path": string  // relative file path to Segmentation mask image file

        }

    ]

3D Box

"3dbox":[

        {

            "box":[x, y, z, dx, dy, dz],           //float  - values are normalized to image width and height

            "class": string,            // label class name

            "score": float             //  value from 0 to 1 in float

        }

    ]

GPS

"gps":[

        {

            "lat": float

            "long": float

        }

    ]

IMU

"imu":[

        {

            "yaw": float

            "pitch": float

            "roll": float

        }

    ]

PCD

"imu":[

        {

            "path": string  // relative file path to pcd file

        }

    ]

Scales

"scales":[

        {

            "path": string  // relative file path to scales file

        }

    ]

Radar

"radar":[

        {

            "path": string  // relative file path to radar file

        }

    ]

Cube

"cube":[

        {

            "path": string  // relative file path to Cube Data file

        }

    ]

Example file