Custom coco dataset in TAO Colab

[jahanzebabro2001]

Please provide the following information when requesting support.
• Network Type (Resnet18)

Hi,
I am pretty new to AI and TAO toolkit seems great for beginners. I’d like to use my own dataset, which I have in coco format already separated in train/test/val folders, and would like to train the SSD resnet18 notebook in colab-pro with it. Right now it seems to work with kitti format but I read in documentation that COCO was supported. I’d appreciate any guidance on how to get the model running with the coco dataset?

[Morganh]

There is no update from you for a period, assuming this is not an issue anymore. Hence we are closing this topic. If need further support, please open a new one. Thanks

@jahanzebabro2001

I’d appreciate any guidance on how to get the model running with the coco dataset?

Hi, currently only KITTI format label is compatible. There is something wrong in the document.
Please convert COCO format to KITTI format.
You can refer to https://github.com/NVIDIA-AI-IOT/deepstream_tao_apps/blob/release/tao3.0/misc/dev_blog/SOTA/dataset_tools/coco2kitti.py.