Skip to content

Latest commit

 

History

History
80 lines (51 loc) · 3.3 KB

Dataset.md

File metadata and controls

80 lines (51 loc) · 3.3 KB
Raw

Dataset for Traffic Sign Recognition

In NXP AIM 2024, we will be working on following traffic signs:

  1. Go Straight Traffic Sign
  2. Turn Left Traffic Sign
  3. Turn Right Traffic Sign
  4. Stop Traffic sign

{% hint style="danger" %} Teams have to divide the dataset into train/validation/test sub-sets on their own. {% endhint %}

Traffic Sign Designs

{% hint style="info" %} Teams can add more images to the data set as per their needs by refering to images of finalised traffic sign design in below file {% endhint %}

{% file src=".gitbook/assets/AIM_2024/Signs_design.docx" %} Traffic Sign Designs and sizes {% endfile %}

Unlabbled data set

The below file conatins all the images that can be used in training of your model are present here.

{% file src=".gitbook/assets/AIM_2024/traffic_sign_dataset_unlabbled.zip" %} Unlabbled data set {% endfile %}

Labbled/bounded data set

The below file conatins all the images along with information about their labels and clases that can be used in training of ypur model are present here.

{% file src=".gitbook/assets/AIM_2024/traffic_sign_dataset_for_pytorch.zip" %} Labbled data set for Pytorch {% endfile %}

{% file src=".gitbook/assets/AIM_2024/Traffic_sign_dataset_yolov5-obb.zip" %} Labbled data set for OBB (Tensorflow) {% endfile %}

{% hint style="warning" %} Teams need to devide the images into train,val and test categories themselves and then create suiatble folders and edit this yaml file {% endhint %}

Labbled/bounded data set (Advance: multi class images)

The below links contain file containing images with multiple signs classes in a single image along with information about their labels and clases that can be used in training of ypur model are present here.

Labbled data set for Pytorch - Pytorch_Dataset_Multi_Class_Images.

Labbled data set for OBB (Tensorflow) - OBB_Dataset_Multi_Class_Images.

{% hint style="warning" %} Teams need to devide the images into train,val and test categories themselves and then create suiatble folders and edit this yaml file {% endhint %}

Unlabbled data set for Traffic Light

The below links conatins all the images that can be used in training of your model to detect the Traffic Light to be used in AIM Grand finale.

Unlabbled data set for red/green light - Traffic_Light_Imahes.

{% hint style="warning" %} The link contains 1700+ red lights and 700+ green light images. We recommend all participants only to train model for red traffic light and consider it an extension of stop sign. {% endhint %}

Labbled data set for Traffic Signs and Traffic Lights

The below links conatins all the images that can be used in training of your model to detect the Traffic Light as well as traffic lights to be used in AIM Grand finale.

Final labbled data set - Final Dataset.

{% hint style="warning" %} This is updated dataset for traffic signs as well as traffic lights. In this data set there are more images, as well as the bounding boxes are improved to achieve better results {% endhint %}