PathDet Dataset Curation - ROADWork Dataset

[m-zain-khawaja]

Description

Parse the ROADWork dataset which can be downloaded from this link to create a dataset comprising input images in PNG format and a single drivable path as the ground truth, based on the trajectory annotations provided in the dataset.

Please create a script called process_roadwork.py in this folder which is responsible for creating the ground truth and saving the data.

The ground truth drivable path information should be stored as a list keypoints defining the path in a JSON format. The drivable path keypoints should be sampled with a vertical (y image axis) displacement of 10 pixels to match the ground truth format used in other datasets.

Please ensure that keypoints are stored in relative coordinates, where the top left-most corner of the image is 0,0 and the bottom right-most corner of the image is 1,1, and all other coordinates are floating point values in the range (0,1) for x,y directions.

Please ensure ground truth images are stored in PNG format

Please also save a drivable path mask which is drawn upon the input RGB image for data auditing purposes as well as a binary drivable path mask in PNG format, where drivable path pixels are assigned a value of 255, and non-drivable path pixels are assigned a value of 0.

Data Summary:

• RGB image in PNG Format
• Drivable path keypoints in JSON Format
• Binary Drivable Path Mask in PNG format
• Drivable Path Mask draw on top of RGB image in PNG format (not used during training, only for data auditing purposes)

[m-zain-khawaja]

@docjag - I have created a new branch for this development called 'pathdet-dataset-curation-roadwork-dataset' - please fork this repo to your github, push the changes on this new branch on your github and then submit the pull request once completed, for review. Thanks!