10 lines
214 B
Python
10 lines
214 B
Python
import onnx
|
|
|
|
from config import MODEL_PATH
|
|
|
|
model_path = MODEL_PATH / 'ssd-mobilenet.onnx'
|
|
model = onnx.load(str(model_path))
|
|
|
|
print(onnx.checker.check_model(model))
|
|
print(onnx.helper.printable_graph(model.graph))
|