im not sure which format your label data is coming in as… but I am personally creating datasets using the COCO format… then overloading the library COCO dataloader with my transforms and paths to images etc. etc…