ssd.pytorch icon indicating copy to clipboard operation
ssd.pytorch copied to clipboard

How to run on custom dataset(COCO FORMAT)

Open caihaunqai opened this issue 4 years ago • 1 comments

Hello, I am a new guy. Anyone who run on custom dataset(COCO FORMAT) can help me? Thank you!!!!

caihaunqai avatar May 29 '21 03:05 caihaunqai

You'd need to write a custom dataset class. The important functions for custom dataset classes are len and getitem, so make sure you've got those in there and you can use the CocoDetection class in data/coco.py as a guide if you get stuck. Good luck!

borschy avatar Feb 23 '22 10:02 borschy