search

varunagrawal / tiny-faces-pytorch

Finding Tiny Faces in PyTorch
MIT License
164 stars 44 forks source link

Train on Custom Dataset #6

Closed MaddyThakker closed 4 years ago

MaddyThakker commented 4 years ago

Is there any direct procedure that I need to follow to train the model on custom Dataset. Which particular files do I need to change?

varunagrawal commented 4 years ago

You would just need to add a new dataloader class (look at the COCO dataloader class for a general idea of how to do this), and you would need to update the num_objects and num_templates for the model.