Import Pytorch model in ONNX format

SameDiff
1

Hi , I have a YOLOv8 Pytorch Model that I converted to ONNX format to be imported but I am hitting this error

image
image1137×770 44.4 KB

Is there something I can do to fix this or is it not supported??

2

@minyueyong what version is this? This should be fixed in the newest master version already. I’m a bit behind with the cuda side of this (technical debt) but we should be good to go soon there.

5

samediff-import-onnx
version: 1.0.0-M2

6

@minyueyong could you try with the latest version? (M2.1)

8

I have added the dependency with the latest version but there is a new error

image
image1064×521 39.7 KB

9

@minyueyong can you link me the model and I’ll try to make sure it works with current release? Thanks!

11

https://concordess-my.sharepoint.com/personal/minyue_yong_gswhouse_com/_layouts/15/onedrive.aspx?id=%2Fpersonal%2Fminyue_yong_gswhouse_com%2FDocuments%2FResearch%2FTOD%2FYOLOv8%20Table%20Object%20Detection&ga=1