add scripts for conversion
Browse files- scripts/tflite_convert.py +15 -0
scripts/tflite_convert.py
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from transformers import TFMobileViTForSemanticSegmentation
|
| 2 |
+
import tensorflow as tf
|
| 3 |
+
|
| 4 |
+
if __name__ == '__main__':
|
| 5 |
+
model = TFMobileViTForSemanticSegmentation.from_pretrained(".")
|
| 6 |
+
converter = tf.lite.TFLiteConverter.from_keras_model(model)
|
| 7 |
+
converter.optimizations = [tf.lite.Optimize.DEFAULT]
|
| 8 |
+
converter.target_spec.supported_ops = [
|
| 9 |
+
tf.lite.OpsSet.TFLITE_BUILTINS,
|
| 10 |
+
tf.lite.OpsSet.SELECT_TF_OPS,
|
| 11 |
+
]
|
| 12 |
+
tflite_model = converter.convert()
|
| 13 |
+
tflite_filename = "tflite_model.tflite"
|
| 14 |
+
with open(tflite_filename, "wb") as f:
|
| 15 |
+
f.write(tflite_model)
|