I am trying to apply the Spatial Transformer Network in a mobile environment, but I don't understand how I am supposed to export this kind of model in order to convert it to TFLITE.
I succesfully adapted TL example tutorial_spatial_transformer_network_dynamic.py and trained that STN network with my own dataset. After that I tried to save it to h5 format using the TL API, but it just saves the weights and not the network structure.
Hello,
I am trying to apply the Spatial Transformer Network in a mobile environment, but I don't understand how I am supposed to export this kind of model in order to convert it to TFLITE.
I succesfully adapted TL example tutorial_spatial_transformer_network_dynamic.py and trained that STN network with my own dataset. After that I tried to save it to h5 format using the TL API, but it just saves the weights and not the network structure.
Can you please provide some advice?
Thank you,