kentaroy47/efficientdet.pytorch

Question training with own dataset and FPS

Opened this issue · 1 comments

Hi Kentaroy,
Thank you for your great job.

  • I would like train my own dataset. Do I change my dataset to COCO and there is different to modify the scale anchor?
  • Do you have train with D1 and D2 and how is the result of FPS ans score?

Thank you again.

@wenjun90
You can either prepare your dataset format as in pascal VOC or COCO.
please see https://github.com/kentaroy47/efficientdet.pytorch/blob/master/train_efficientdet.ipynb
I haven't tried D1, D2 yet, but it is sure on top of my list.