Code Monkey home page Code Monkey logo

m-lsd-onnxrun-cpp-py's Introduction

M-LSD-onnxrun-cpp-py

使用ONNXRuntime部署面向轻量实时的M-LSD直线检测,包含C++和Python两个版本的程序

本套程序里在weights文件夹里有4个onnx模型文件,每个模型文件的大小不超过10M。 经过运行程序实验比较,model_512x512_large.onnx的检测效果最好。

本套程序对应的paper是顶会AAAI2022里的一篇文章《Towards Light-weight and Real-time Line Segment Detection》

m-lsd-onnxrun-cpp-py's People

Contributors

hpc203 avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar

m-lsd-onnxrun-cpp-py's Issues

onnx的转化

冒昧的打扰:
用您给的onnx文件可以顺利的进行img的推导
然后用官方代码 https://github.com/lhwcv/mlsd_pytorch.git 用trt_converter.py 进行onnx的转化可顺利进行, 同时
onnx_model = onnx.load(out_file), onnx.checker.check_model(onnx_model)检查没有问题.
但是加载自己转化后的onnx文件,加载阶段出现问题如下:

    self.input_name = self.onnx_session.get_inputs()[0].name
    print(" input name is: ", self.input_name)
    #input_name: input_image_with_alpha:0
    #not good: pading0

    self.output_names = [self.onnx_session.get_outputs()[i].name for i in range(3)]
    print(" output name is: ", self.output_names)
    # ['Identity', 'Identity_1', 'Identity_2']
    #not good: list index out of range

我该如何修改,或者怎么得到您的帮助

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.