search

marcelsheeny / radiate_sdk

SDK developed to access data from RADIATE dataset
MIT License
94 stars 34 forks source link

About dataloader #1

Closed xllau closed 3 years ago

xllau commented 3 years ago

Hi, would you like to provide the training and validation dataloader? Since there is not anything find in your SDK.

marcelsheeny commented 3 years ago

I added the training code https://github.com/marcelsheeny/radiate_sdk/blob/master/vehicle_detection/train.py

This code has the logic for both horizontal and oriented bounding boxes, however it is not working with oriented bounding boxes since the current version of detectron2 doesn't support training with rotated bboxes (AFAIK).

xllau commented 3 years ago

I added the training code https://github.com/marcelsheeny/radiate_sdk/blob/master/vehicle_detection/train.py

This code has the logic for both horizontal and oriented bounding boxes, however it is not working with oriented bounding boxes since the current version of detectron2 doesn't support training with rotated bboxes (AFAIK).

Thanks a lot for your update! And I have applied for your dataset access for few days, hope I can get it sooner!

marcelsheeny commented 3 years ago

We are an analysing each request individually and we will give you soon access to the dataset.