Question training with own dataset and FPS

kentaroy47 / efficientdet.pytorch

:neckbeard:Unofficial implementation of EfficientDet

MIT License

61 stars 16 forks source link

Question training with own dataset and FPS #9

Open wenjun90 opened 4 years ago

wenjun90 commented 4 years ago

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.

kentaroy47 commented 4 years ago

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