Skip to content

Convert pytorch to onnx #5

@danielzhangau

Description

@danielzhangau

我在原代码model/detector.py的forward function里修改,因为我看这里有两个tensor output,但他们size不一样
image

image
我也不是很确定是不是修改这里,求指导~ Any helps will be appreciated!!

Originally posted by @danielzhangau in #1 (comment)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions