Obtain ONNX Input Size| 东毅居士

Obtain ONNX Input Size

作者：XD / 发表： 2021年9月25日 08:18 / 更新： 2021年9月25日 08:18 / 编程笔记 / 阅读量：2442

Obtain ONNX Input Size

import onnx

path = "onnx_model.onnx"
model = onnx.load(path)
input_shape = model.graph.input[0].type.tensor_type.shape.dim
height = int(str(input_shape[2]).strip().split(': ')[-1])
width = int(str(input_shape[3]).strip().split(': ')[-1])

本文作者：XD 转载请标明出处：http://www.eadst.com/blog/58

本站采用知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议进行许可。

上一篇
Python Numpy Array to C++ Vector

下一篇
Convert Math Equation from Picture to LaTeX

Category

标签云

CV TensorRT Vmess HuggingFace v2ray Linux Streamlit Knowledge Qwen2.5 Pickle Base64 Safetensors diffusers CAM RGB FP32 Plate scipy 域名 NLTK Magnet Bin GIT Git ResNet-50 EXCEL Windows Plotly Bitcoin Augmentation llama.cpp Gemma InvalidArgumentError Color LeetCode SVR VPN QWEN UNIX Bipartite 签证 git Hotel Django NLP Food LLAMA tar Interview Attention 搞笑 Anaconda PDF TSV Proxy Clash YOLO Michelin OpenCV PIP Web transformers CEIR CSV Paddle HaggingFace VGG-16 关于博主 BF16 TensorFlow CTC Password XGBoost FP64 Dataset FP16 LaTeX Statistics Shortcut Land Hilton Video Google uwsgi OpenAI JSON Input Baidu GGML 算法题 Website Quantization Random Conda 阿里云 Data FastAPI BTC Python Card Pillow Qwen 腾讯云 PyCharm BeautifulSoup Nginx Tiktoken C++ GPT4 GPTQ Image2Text OCR GoogLeNet Domain torchinfo Distillation Pytorch Quantize FlashAttention Template Hungarian VSCode Crawler Use Vim XML LLM Algorithm git-lfs Tensor Heatmap Sklearn AI Tracking Pandas Ptyhon COCO NameSilo Excel SQL Zip LoRA Markdown DeepStream Bert WebCrawler uWSGI ChatGPT Diagram Permission 报税 Translation Paper logger Math SQLite Docker CUDA Ubuntu Disk Datetime Jetson Transformers ModelScope Cloudreve Numpy Review MD5 hf SPIE ONNX Github

站点统计

本站现有博文266篇,共被浏览440639次

本站已经建立2019天!

原 Obtain ONNX Input Size

作者：XD / 发表： 2021年9月25日 08:18 / 更新： 2021年9月25日 08:18 / 编程笔记 / 阅读量：2442

Obtain ONNX Input Size