search

naseemap47 / YOLO-NAS

Train and Inference your custom YOLO-NAS model by Single Command Line
Apache License 2.0
97 stars 13 forks source link

custom Dataloaders #11

Closed kaustubhroy1995 closed 1 year ago

kaustubhroy1995 commented 1 year ago

Hi

I would like to get a peek at the structure of the dataloader used/the format in which the model expects input and write a custom dataloader in the format that supports the Trainer class. Could you link me to a code/guide on how to write my own dataloader?

naseemap47 commented 1 year ago

I don't know how to write a custom data-loader. I used the function from super_gradients library for loading my data.

from super_gradients.training.dataloaders.dataloaders import coco_detection_yolo_format_train, coco_detection_yolo_format_val
naseemap47 commented 1 year ago

I don't know how to write a custom data-loader. I used the function from super_gradients library for loading my data.

from super_gradients.training.dataloaders.dataloaders import coco_detection_yolo_format_train, coco_detection_yolo_format_val

I think this will help you. So I am closing this issue. If you have any questions please let me know and re-open this issue. Thank you.