search

martinsbruveris / tensorflow-image-models

TensorFlow port of PyTorch Image Models (timm) - image models with pretrained weights.
https://tfimm.readthedocs.io/en/latest/
Apache License 2.0
287 stars 25 forks source link

[FEATURE REQUEST] EfficientNet models #65

Closed benihime91 closed 2 years ago

benihime91 commented 2 years ago

Is there any plan to add efficientnet V1x or V2x models ? I know implementations can be found in the keras module itself but adding these models would make this library trully the equivalent of ross's pytorch-image-models.

martinsbruveris commented 2 years ago

Yes, there is a plan to port the efficientnet models from timm as well. It is a bigger chunk of work. If all goes well, I hope to have a new version of tfimm with efficientnet models published in April... 🤞 ...

benihime91 commented 2 years ago

Great