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
286 stars 25 forks source link

[SAM] make saving to TF saved_model work #81

Closed freedomtan closed 1 year ago

freedomtan commented 1 year ago

I was trying to save SAM models to TF saved_model (and converting to TFLite later) but failed to do for sam.mask_decoder and sam.image_encoder.

martinsbruveris commented 1 year ago

Thank you! Taking a look at the PR. Will add a unit test and merge.

martinsbruveris commented 1 year ago

Thanks for your contribution! Much appreciated!