google-research / big_transfer

Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
https://arxiv.org/abs/1912.11370
Apache License 2.0
1.5k stars 175 forks source link

Object detection pre-trained weights #20

Open arunponnusamy opened 4 years ago

arunponnusamy commented 4 years ago

First of all kudos to the great work.

In the official blog post on Google AI blog, it is mentioned that BiT models pre-trained on large datasets perform better on object detection task as well. But in the repo couldn't find any details about object detection models.

Any plans to release object detection BiT models and tutorial/colab for fine tuning on custom dataset ?

Cheers.

Zumbalamambo commented 4 years ago

+1

lucasb-eyer commented 4 years ago

Thanks for your interest! We are unlikely to release that as it was trained in a non-public code-base, but in principle you should be able to, for example, plug the released PyTorch BiT models into Detectron2 or similar and get a corresponding boost in performance.