تنزيل onnx simplifier - تنزيل رمز مصدر onnx simplifier

onnx simplifier

بايثون

1.0.0

تنزيل

onnx simplifier

Onnx رائع ، ولكن في بعض الأحيان معقدة للغاية.

خلفية

في يوم من الأيام أردت تصدير عملية إعادة تشكيل بسيطة التالية إلى Onnx:

 import torch


class JustReshape ( torch . nn . Module ):
    def __init__ ( self ):
        super ( JustReshape , self ). __init__ ()

    def forward ( self , x ):
        return x . view (( x . shape [ 0 ], x . shape [ 1 ], x . shape [ 3 ], x . shape [ 2 ]))


net = JustReshape ()
model_name = 'just_reshape.onnx'
dummy_input = torch . randn ( 2 , 3 , 4 , 5 )
torch . onnx . export ( net , dummy_input , model_name , input_names = [ 'input' ], output_names = [ 'output' ])

شكل الإدخال في هذا النموذج ثابت ، لذلك ما كنت أتوقع

Simple_reshape

ومع ذلك ، حصلت على النموذج المعقد التالي بدلاً من ذلك:

المعقد

حلنا

يتم تقديم OnNx Simplifier لتبسيط نموذج ONNX. يثير الرسم البياني للحساب بأكمله ثم يحل محل المشغلين الزائدين بمخرجاتهم الثابتة (ويعرف أيضًا باسم طي ثابت).

إصدار الويب

لقد نشرنا OnNx Simplifier على convertModel.com. إنه يعمل خارج الصندوق ولا يحتاج إلى أي تثبيت . لاحظ أنه يعمل في المتصفح محليًا وأن نموذجك آمن تمامًا.

نسخة بيثون

 pip3 install -U pip && pip3 install onnxsim

ثم

 onnxsim input_onnx_model output_onnx_model

للحصول على المزيد من الميزات المتقدمة ، جرب الأمر التالي للحصول على رسالة مساعدة

 onnxsim -h

توضيح

مقارنة عامة بين النموذج المعقد ونسخته المبسطة:

مقارنة بين النموذج القديم والنموذج الجديد

سير العمل في النصوص

إذا كنت ترغب في تضمين حزمة Onnx Simplifier Python في نص آخر ، فهذا بسيط.

 import onnx
from onnxsim import simplify

# load your predefined ONNX model
model = onnx . load ( filename )

# convert model
model_simp , check = simplify ( model )

assert check , "Simplified ONNX model could not be validated"

# use model_simp as a standard ONNX model object