Skip to content

如何导出NAFNetLocal模型文件为onnx格式? #857

Description

@lmw0320

27acc3baaef9102b2a7372d1c6003776

如图所示,我尝试将这个模型文件导出onnx文件,实际设置为动态尺寸输入时,会报错:
1721721548726

1721721376005

我检查了NAFNet网络中的check_image_size函数,并增加了打印信息,用于查验:
1721721721799

不知道这个模型,到底要如何导出呢??
----我个人测试过,如果模型事先设定一张图片的尺寸数据,导出onnx文件后。可以正常调用onnx,对该图片进行 推理,结果也正常。而使用另外一张不同尺寸的图片进行onnx模型推理时,就会报错,提示输入的数据形状不对。。---因此,我才希望导出一个可以支持动态尺寸输入的onnx文件,以便进行正常应用。

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions