search

rwightman / efficientdet-pytorch

A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights
Apache License 2.0
1.58k stars 293 forks source link

[FEATURE] OpenImages Pretrained Models #216

Open rsomani95 opened 3 years ago

rsomani95 commented 3 years ago

Hi Ross, thanks for sharing this repo! Curious if you have any plans on releasing a model trained on OpenImages?

Is your feature request related to a problem? Please describe. No

Describe the solution you'd like Have access to a pretrained model trained on OpenImages

Describe alternatives you've considered The official repo (see here), but their model is outdated and it's also in tensorflow, which is inconvenient

Additional context N/A

rwightman commented 3 years ago

@rsomani95 that has been the goal, I did train one once (> 1 year ago) on a faster-rcnn type net w/ first version of my dataset code, but not with these nets yet. It takes a bloody long time so I've been waiting until I have more access to more compute. I might be able to do it on TPUs soon, but getting this working on PyTorch XLA could be a bit of work.

rsomani95 commented 3 years ago

@rwightman Thanks for the response. Great to know it's somewhere on your roadmap. I'd love to have helped but have no experience PyTorch XLA and am unlikely to gain access to TPUs anytime soon.

jdsannchao commented 2 years ago

Hi Ross, Thanks for sharing this repo! May I ask how is the plan going, were you able to train the model on OpenImages?

rwightman commented 2 years ago

@jdsannchao no not yet, I only have 2x GPU machines free right now and it'd probably take close to two months for a decent sized model. Getting closer to the point where I think PyTorch XLA might work though... but have more to do still