onnx 파일의 shape를 확인하는 방법입니다. 2023. 11. 5 최초작성 참고 https://stackoverflow.com/questions/56734576/find-input-shape-from-onnx-file >>> import onnxruntime as ort >>> model = ort.InferenceSession("bytetrack_s.onnx", providers=['CUDAExecutionProvider', 'CPUExecutionProvider']) C:\Users\jeong\miniconda3\envs\bytetrack\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py:53: UserWarning: ..

Keras 모델( .h5)을 onnx로 변환한 후, OpenCV DNN에서 onnx 모델을 사용해서 추론을 해봅니다. 2022. 10. 25 최초작성 1. MNIST 데이터셋에 대한 모델을 학습시켜 h5 모델 파일로 저장합니다. 코드 출처 https://www.tensorflow.org/tutorials/quickstart/beginner?hl=ko import tensorflow as tf mnist = tf.keras.datasets.mnist (x_train, y_train), (x_test, y_test) = mnist.load_data() x_train, x_test = x_train / 255.0, x_test / 255.0 model = tf.keras.models.Sequential([ t..