batchgenerators icon indicating copy to clipboard operation
batchgenerators copied to clipboard

DataLoader for object detection

Open lspinheiro opened this issue 5 years ago • 3 comments

Is there an example of what data structure and key names should we use for object detection labels? I'm trying to use this together with medicaldetectiontoolkit and I couldn't find an object detection example pipeline, only for segmentation.

lspinheiro avatar Jan 03 '20 01:01 lspinheiro

Hi, I am not knowledgeable enough for this, but I know that somebody in our department is currently working on this. I will try to get in tough with them. @pfjaeger Do you have any examples for how to use batchgenerators together with the medicaldetectiontoolkit? Best, Fabian

FabianIsensee avatar Jan 31 '20 10:01 FabianIsensee

Hi, the medicaldetectiontoolkit inherently uses the batchgenerators, and all examples are for object detection. I am not sure I understand your question, did you see the example dataloaders ?

pfjaeger avatar Jan 31 '20 12:01 pfjaeger

Hi Paul,

I did see the examples, I ran the toy example and it basically generated mask layers for images (using various shapes). I didn't run the other example because I didn't have the dataset.

But is this the expected input even for object detection? A mask with the bounding box?

lspinheiro avatar Jan 31 '20 22:01 lspinheiro