9 lines
241 B
Python
9 lines
241 B
Python
|
import onnx
|
||
|
|
|
||
|
|
model = onnx.load('oak_d_deployment/yolox_model.onnx')
|
||
|
|
print('Inputs:')
|
||
|
|
for inp in model.graph.input:
|
||
|
|
print(f' {inp.name}: {inp.type}')
|
||
|
|
print('Outputs:')
|
||
|
|
for out in model.graph.output:
|
||
|
|
print(f' {out.name}: {out.type}')
|