search

Media-Smart / vedadet

A single stage object detection toolbox based on PyTorch
Apache License 2.0
498 stars 128 forks source link

How to train with custom dataset by using the pretrained model? #94

Open SwEngine opened 1 year ago

SwEngine commented 1 year ago

Hello, I would like to ask a few questions about this github repo.

  1. What do I need to do to train using the pretrained model?
  2. How can I create my own custom dataset other than Wider-Face? Is there an annotation tool you recommend that does annotation in the same format?
  3. What should I do to train with the dataset I created?
  4. How can I convert the model to onnx after training?
gdiasbruno commented 1 year ago

Did you manage to do it?