ssd.pytorch
ssd.pytorch copied to clipboard
How to run on custom dataset(COCO FORMAT)
Hello, I am a new guy. Anyone who run on custom dataset(COCO FORMAT) can help me? Thank you!!!!
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!