Model conversion from Pytorch to Tf using Onnx.

• nnUNet

  11 5,019 9.0 Python
  

Hi all, Does anyone have experience with converting a nnUNet pytorch models into Tf using Onnx?

• mmpose

  31 5,002 8.0 Python

  OpenMMLab Pose Estimation Toolbox and Benchmark.

  

I downloaded pytorch2onnx.py from mmPose tools. It's big, but the top half is imports and input arguments. Line 125, I hard-coded my (image) input size. I ran it on my .pth model file, and out pop'd an onnx file.

• InfluxDB

  www.influxdata.com sponsored

  Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.

  

Suggest a related project