Files
sunnypilot/selfdrive/modeld/test/tf_test/pb_loader.py
Willem Melching 2ac27def5a Running pre-commit in CI (#1590)
* Running pre-commit in CI

* fix dockerfile syntax

* dont run on submodule repo folders

* Fix some import errors in ci

* more stuff

* That should be the last one
old-commit-hash: ba2ac1207a
2020-05-28 15:40:48 -07:00

9 lines
258 B
Python
Executable File

#!/usr/bin/env python3
import sys
import tensorflow as tf # pylint: disable=import-error
with open(sys.argv[1], "rb") as f:
graph_def = tf.compat.v1.GraphDef()
graph_def.ParseFromString(f.read())
#tf.io.write_graph(graph_def, '', sys.argv[1]+".try")