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

Question training with own dataset and FPS

Open wenjun90 opened this issue 5 years ago • 1 comments
trafficstars

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 avatar May 14 '20 15:05 wenjun90

@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.

kentaroy47 avatar May 14 '20 23:05 kentaroy47