Any script to train the model ?

[Samjith888]

Hi @ChristianMarzahl

Thank you for your great work, I'm really excited to implement this model with my dataset. I couldn't find any script for training the model. I have a kitti dataset, and i just converted into COCO, How can i train by using this model?

[ChristianMarzahl]

Moin,

please take a lock at: https://github.com/ChristianMarzahl/ObjectDetection/blob/master/examples/CocoTiny_Retina_Net.ipynb

That model is trained with the COCO format.

[Samjith888]

Moin,

please take a lock at: https://github.com/ChristianMarzahl/ObjectDetection/blob/master/examples/CocoTiny_Retina_Net.ipynb

That model is trained with the COCO format.

I got following error,

    from BoundingBox import BoundingBox
ModuleNotFoundError: No module named 'BoundingBox'

I found the solution here i just cloned and added into the sys path , but still getting the same error

[ChristianMarzahl]

Please make sure that you added the lib folder to your system path "****/Object-Detection-Metrics/lib"

[Samjith888]

Please make sure that you added the lib folder to your system path "****/Object-Detection-Metrics/lib"

My custom dataset is in kitti format. I have a python script which read those kitti format files and convert into the following format (COCO).

[{'file_name': '/home/samjith/0000180.jpg', 'height': 788, 'width': 1400, 'image_id': 1, 'annotations': [{'bbox': [250.0, 675.0, 23.0, 17.0], 'bbox_mode': <BoxMode.XYWH_ABS: 1>, 'area': 391.0, 'segmentation': [], 'category_id': 0}, {'bbox': [295.0, 550.0, 21.0, 20.0], 'bbox_mode': <BoxMode.XYWH_ABS: 1>, 'area': 420.0, 'segmentation': [], 'category_id': 0},..

Where should i pass this data_dict in your script?