机器学习
captain_richard
get the perl!!!!
展开
-
nvidia 各种框架(如pytorch)训练出现的奇怪错误
1 ubuntu16.04 cuda10.2 nvidia-driver:470.74 docker镜像 多gpu节点pytorch训练yolov5,出现RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR,1.1 初步判断是多节点训练引起,,改成单节点,调节batchsize发现小batchsize才不会出错,,但是显卡还没有很多显存,判断是显卡驱动版本引起,因为该镜像在其他机器不会出现这种情况,,把470换成450版本显卡驱动,上面问题消...原创 2021-10-14 13:48:48 · 245 阅读 · 0 评论 -
onnxrutime api
1 遇到session.run函数运行一段时间会出现segment fault,不定时出现,可能是没把Qrt::Env这个变量放入整个程序的生命周期,所以尽量不要把Qrt::Env弄成临时变量,应该让其伴随所需使用的生命周期,可以声明成类的成员或者全局变量...原创 2021-03-19 19:57:43 · 407 阅读 · 2 评论 -
Yolo v5 pytorch模型转onnx用c++进行推理
yolov5 c++ python java 推理原创 2021-03-02 14:16:45 · 9452 阅读 · 16 评论 -
ncnn技巧
1 开启加速https://zhuanlan.zhihu.com/p/112564372原创 2020-11-06 14:41:27 · 188 阅读 · 0 评论 -
numpy 与opencv svd的不同
最近用到奇异值分解,即SVD,numpy 和opencv中都有svd分解的api,但是得出来的结果稍有差异,但是数值上差异基本不大,主要是符号上有些不同numpy里的api是 numpy.linalg.svd(mat),得出的顺序是U,S,Vopencv里的api是cv2.SVDecomp(mat)得出的顺序是S,U,V,顺序别搞混对于S值,两者数值一样对于U值,两者在第一列的数值符号是相反的对于V值,两者在第一行的数值符号是相反的...原创 2020-09-05 16:15:43 · 1245 阅读 · 0 评论 -
Opencv 使用中的注意事项
cv::Mat 除了常规的初始化深浅拷贝,注意从指针或数组初始化mat时,记住改初始化相当于浅拷贝原创 2020-05-09 17:45:42 · 191 阅读 · 0 评论 -
tensorlfow c++库 windows使用 问题汇总
1.可以去https://github.com/fo40225/tensorflow-windows-wheel下载编译好的tensorflow 库2.如果执行程序出现类似缺少cublas64xxx.dll这样的错误,大概率是cuda库问题,先看电脑有无安装cuda库,可以去nvidia官网下载cuda tookit库,二是看你下载的tensorflow c++库所用的cuda版本与下载安装...原创 2019-09-08 11:12:34 · 251 阅读 · 0 评论