Giter Club home page Giter Club logo

ultra-fast-lane-detection-v2-opencv-onnxrun's Introduction

Ultra-Fast-Lane-Detection-v2-opencv-onnxrun

分别使用OpenCV、ONNXRuntime部署Ultra-Fast-Lane-Detection-v2车道线检测,包含C++和Python两种版本的程序。

Ultra-Fast-Lane-Detection-v2是TPAMI2022期刊里的论文,它是速度精度双SOTA的最新车道线检测算法。

本套程序提供了分别在culane和tusimple数据集训练出来的模型,onnx模型文件在 百度云盘,链接:https://pan.baidu.com/s/1b68-q_NX2PQPHZQn2h-x2A 提取码:jfwn

需要注意的是opencv版本需要是4.7以上的,opencv-dnn才能正常读取onnx文件做推理的

ultra-fast-lane-detection-v2-opencv-onnxrun's People

Contributors

hpc203 avatar

Stargazers

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

Watchers

 avatar  avatar

ultra-fast-lane-detection-v2-opencv-onnxrun's Issues

代码中的resize与裁剪crop问题

AI工程师你好,tusimple-onnx的尺寸为(320,800),为什么项目例onnxruntime的推理detect部分先将图片resize到(400,800)然后裁剪从最后一行往上数320行,得到(320,800)的图片呢,直接resize到(320,800)会报错我试了,为什么这样呢,我先resize到其他比例可以吗,有什么原则吗

使用opencv4.7报错

outnames :
exist_col
exist_row
loc_col
loc_row
[ERROR:[email protected]] global net_impl.cpp:1164 cv::dnn::dnn4_v20221220::Net::Impl::getLayerShapesRecursively OPENCV/DNN: [Reshape]:(onnx_node!Reshape_48): getMemoryShapes() throws exception. inputs=1 outputs=1/1 blobs=0
[ERROR:[email protected]] global net_impl.cpp:1167 cv::dnn::dnn4_v20221220::Net::Impl::getLayerShapesRecursively input[0] = [ 1 8 10 25 ]
[ERROR:[email protected]] global net_impl.cpp:1171 cv::dnn::dnn4_v20221220::Net::Impl::getLayerShapesRecursively output[0] = [ 1 4000 ]
[ERROR:[email protected]] global net_impl.cpp:1177 cv::dnn::dnn4_v20221220::Net::Impl::getLayerShapesRecursively Exception message: OpenCV(4.7.0) D:\tools\opencv\opencv-4.7.0\modules\dnn\src\layers\reshape_layer.cpp:150: error: (-1:Backtrace) Can't infer a dim denoted by -1 in function 'cv::dnn::computeShapeByReshapeMask'

OpenCV: terminate handler is called! The last OpenCV error is:
OpenCV(4.7.0) Error: Backtrace (Can't infer a dim denoted by -1) in cv::dnn::computeShapeByReshapeMask, file D:\tools\opencv\opencv-4.7.0\modules\dnn\src\layers\reshape_layer.cpp, line 150

无法使用自己转换的ONNX推理

大佬你好,感谢您的开源工作!我在使用你提供ufldv2_culane_res18_320x800.onnx文件在main.py推理时成功,但在使用我自己转换的culane_res18.onnx文件无法成功,出现如下报错:(culane_res18.onnx是利用官方权重转换而来)
/home/pc/anaconda3/envs/UFLD/bin/python3.7 /home/pc/FY/Ultra-Fast-Lane-Detection-v2-master/ONNX/opencv/main.py
Traceback (most recent call last):
File "/home/pc/FY/Ultra-Fast-Lane-Detection-v2-master/ONNX/opencv/main.py", line 99, in
net = Ultra_Fast_Lane_Detection_v2(args.modelpath)
File "/home/pc/FY/Ultra-Fast-Lane-Detection-v2-master/ONNX/opencv/main.py", line 10, in init
self.input_height = int(input_shape[0])
ValueError: invalid literal for int() with base 10: 'res18'

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.