search

toandaominh1997 / EfficientDet.Pytorch

Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch
MIT License
1.44k stars 305 forks source link

EfficientDet-D3-4(with Weight) #133

Open Jilliansea opened 4 years ago

Jilliansea commented 4 years ago

Where can I found EfficientDet-D3-D7(with Weight)? Thank you

EdwardMessi commented 4 years ago

I also want to know..It's so difficult to train this all by myself.

zhui-ying commented 4 years ago

Can I get your pre-train model?

tadbeer commented 4 years ago

Hi @EdwardMessi , @Jilliansea I am trying to train this on my own dataset, but cant achieve a successfull training, that is, the MAP is not increasing with increasing epochs. Have you tried training on your own dataset? Could you kindly get in touch with me?

zylo117 commented 4 years ago

Hi @EdwardMessi , @Jilliansea I am trying to train this on my own dataset, but cant achieve a successfull training, that is, the MAP is not increasing with increasing epochs. Have you tried training on your own dataset? Could you kindly get in touch with me?

the backbone is altered in the wrong way, the conv stride of efficientnet should not be changed.

for more info, you can checkout my repo,

https://github.com/zylo117/Yet-Another-EfficientDet-Pytorch

tadbeer commented 4 years ago

Thanks @zylo117 . So can i go ahead and try finetuning on my own data using your repository? Has anyone or you yourself have, reported a successfull finetuning on a self made dataset?

zylo117 commented 4 years ago

Thanks @zylo117 . So can i go ahead and try finetuning on my own data using your repository? Has anyone or you yourself have, reported a successfull finetuning on a self made dataset?

of course, the training script is still testing and will be released within a few days. and yes, i trained on a custom dataset, it worked alright.