- 博客(44)
- 资源 (16)
- 收藏
- 关注
原创 Pytorch使用集成可形变卷积构建网络并导出onnx模型
解决RuntimeError: Exporting the operator deform_conv2d to ONNX opset version 14 is not supported. Please feel free to request support or submit a pull request on PyTorch GitHub问题参考一、安装对应的包pip install deform_conv2d_onnx_exporter二、构建简易网络模型import torchim
2024-09-19 20:26:17
671
1
原创 Pytorch添加自定义算子之(13)-CMake与Ninja编译Pytorch自定义算子
参考CMakeVS20191.3 custom_group_norm.cpp1.4 编译成功2.2 your_extension.cpp2.3 测试文件test.py2.4 结果解决:找到报错处,将’utf-8’改成gbk编码,或者其他编码,如下3.2 windows下直接改成Utf-8即可。
2024-09-09 21:51:25
420
原创 Pytorch添加自定义算子之(12)-开闭原则设计tensorrt和onnxruntime推理语义分割模型
/ 导入动态库__declspec# pragma once # ifndef ABSTRACT_BASE_H # define ABSTRACT_BASE_H // 导入动态库 # ifdef _WIN32 # define IMPORT __declspec(dllimport) # else # define IMPORT # endif # include <opencv/cv.hpp> # include <string> using namespace std;
2024-08-22 20:40:08
243
原创 Pytorch添加自定义算子之(11)-C++应用程序将onnx模型编译并转成tensorrt可执行模型
参考opencvtensorrtcudavisual studio 2019cmake三、trt_dep.cpp文件和trt_dep.hpp文件四、主函数segment.cpp五、CMak配置1.onnx模型需时静态尺寸模型。2.当模型输出为yolov5的浮点型时,需要修改屏蔽trt_dep.cpp处的输出,BiseNet分割则不用。
2024-08-08 21:20:36
244
原创 Pytorch添加自定义算子之(10)-mmdeploy编译流程
pplcv 是 openPPL 开发的高性能图像处理库。此依赖项为可选项,只有在 cuda 平台下,才需安装。
2024-07-02 21:33:10
274
原创 Pytorch添加自定义算子之(9)-mmcv编译流程
添加 C:\Program Files (x86)\Microsoft Visual Studio\2019\Community\VC\Tools\MSVC\14.27.29110\bin\Hostx86\x64 到 PATH,则 cl.exe 可以在命令行中运行。修改D:\anaconda3\envs\mmcv\Lib\site-packages\torch\include\torch\csrc\jit\api\module.h。设置系统编码为UTF-8,只在当前cmd有效。
2024-06-30 19:51:15
274
原创 Pytorch添加自定义算子之(8)-最远点采样自定义算子三种方式验证数据
一、原始pytorch实现最远点采样算法import torchdef farthest_point_sample(xyz, M): """ Input: xyz: pointcloud data, [B, N, 3] M: number of samples Return: centroids: sampled pointcloud index, [B, M]采样点云索引 """ device = xyz.dev
2024-06-30 19:48:06
139
原创 Python收录总结笔记一
主要原因是在单核时候,同时只有一个线程在执行CPU,所以这个线程总能获取到GIL。当转换到多核时,同时会有多个线程在不同的CPU上执行,此时不同的线程之间就需要竞争GIL,而GIL只能同时被一个线程申请到,所以会导致其它线程处于闲置状态。过用eval内置方法可以将一个字符串转成python对象,不过,eval方法是有局限性的,对于普通的数据类型,json.loads和eval都能用,但遇到特殊类型的时候,eval就不管用了,所以eval的重点还是通常用来执行一个字符串表达式,并返回表达式的值。
2024-03-27 23:35:46
700
原创 windows家庭版开启远程方式
下载RDP wrap相关链接为:https://github.com/stascorp/rdpwrap/releases/download/v1.6.2/RDPWInst-v1.6.2.msi解压RDPWrap.zip,以管理员形式运行install.bat选自:https://blog.youkuaiyun.com/qq_41242689/article/details/124715297https://github.com/anhkgg/SuperRDP
2024-03-25 20:12:14
267
原创 python将旋转目标检测到的txt转labelimg2中斜矩形框的xml(正矩形通用)
【代码】python将旋转目标检测到的txt转labelimg2中斜矩形框的xml。
2024-03-15 21:31:57
433
原创 Pytorch可形变卷积分类模型与可视化
models:模型存放路径以及日志存放路径。grad_cam:可视化图存放路径。image:数据集存放路径。
2024-03-10 13:19:49
499
原创 python实现xml标注数据正矩形框的目标检测数据增强并保存新的xml
【代码】python实现xml标注数据正矩形框的目标检测数据增强并保存新的xml。
2024-03-06 14:52:06
437
原创 YOLO5单独提取代码计算预测框与类别的Map0.5-0.95
import numpy as npdef ap_per_class(tp, conf, pred_cls, target_cls, eps=1e-16): """ Compute the average precision, given the recall and precision curves. Source: https://github.com/rafaelpadilla/Object-Detection-Metrics. # Arguments t
2024-03-03 11:13:51
621
原创 python将labelme中的json转labelimg中正矩形框的xml
【代码】python将labelme中的json转labelimg中正矩形框的xml。
2024-03-01 17:12:19
440
原创 Python根据3个点确定两个向量之间的夹角-180度到180方向进行矫正
【代码】Python根据3个点确定两个向量之间的夹角-180度到180方向进行矫正。
2024-02-29 16:41:52
619
原创 Pytorch添加自定义算子之(7)-使用onnxruntime自定义GPU与CPU最远点采样算子加载onnx模型
参考:https://gitcode.net/mirrors/microsoft/onnxruntime/-/releases?spm=1033.2243.3001.5877三、配置cuda_ops.cu(无作用)四、配置cuda_ops.h(无作用)五、配置custom_op_infer.h六、配置custom_op_test.cc七、配置sampling.cpp八、配置sampling_gpu.cu文件九、配置sampling_gpu.h文件十、查看模型输出结果查看GP
2024-02-27 21:33:25
462
原创 Pytorch添加自定义算子之(6)-最远点采样自定义算子添加至pytorch中使用并导出onnx模型
参考:Pytorch2Onnxruntime的github地址,忘记了具体网址了。
2024-02-27 21:32:14
177
原创 Pytorch添加自定义算子之(5)-配置GPU形式的简单add自定义算子
参考:https://zhuanlan.zhihu.com/p/358778742。
2024-02-27 21:25:57
627
原创 Pytorch添加自定义算子之(4)-配置onnxruntime的python调用
参考上一篇C++的调用:https://blog.youkuaiyun.com/baidu_28342107/article/details/132734699参考onnxruntime的windows调用:2.动态库cpp文件:custom_py.cc二、构建CMakeLists.txt三、编译可以看到成功后动态库.so文件以及路径
2024-02-27 21:25:37
334
9
原创 Pytorch添加自定义算子之(3)-配置onnxruntime的C++调用
命名为:custom_op.cc/**/private:public:floatconst/*} };return 1;/*} };
2024-02-27 21:24:57
359
原创 Pytorch添加自定义算子之(2)-配置torch算子
构建py文件名称为:export_custom_op.py可能出现的问题:重新较低版本的:protobuf。
2024-02-27 18:53:32
275
转载 C/C++执行cmd命令-system()
cmd常用命令 calc:启动计算器notepad:打开记事本netstat -a:查看所有的端口tasklist:查看所有的进程d: : 切换到D盘dir(directory) 列出当前目录下的文件以及文件夹md (make directory) 创建目录rd (remove directory) 删除目录(带内容的文件或者文件夹不能直接删除,必须先删除里面,再删除外面)。如果要删除非空目录,...
2020-09-14 00:39:47
2856
转载 数据库初步操作转载
因为今天上午刚学了慕课上哈工大的数据库系统上的嵌入式SQL,然而他只讲了嵌入式sql语句的语法,并没有将如何在编译器上设置,搜了半天才发现这个mysql提供了函数接口代替了那些嵌入式SQL语句,所以又转而去在vs上配置MySQL。首先声明一下我用的是vs2010和mysql5.5,这个和mysql的图形化工具没关系,只需要编译器和mysql就行了,下面是配置的步骤:1.创建一个项目:创建项目:创建一个空项目初始代码:创建一个源文件(用来测试是否连接成功并),内容如下:我第一次看也是一脸懵逼,怎么和
2020-08-21 23:17:14
201
原创 机器学习之BP神经网络算法实现图像分类
BP 算法是一个迭代算法,它的基本思想为:(1) 先计算每一层的状态和激活值,直到最后一层(即信号是前向传播的);(2) 计算每一层的误差,误差的计算过程是从最后一层向前推进的(这就是反向传播算法名字的由来);(3) 更新参数(目标是误差变小),迭代前面两个步骤,直到满足停止准则(比如相邻两次迭代的误差的差别很小)。下面用图片的形式展示其推到过程数据集:数据集采用Sort_1000pics数据...
2018-10-23 16:16:30
24004
13
Texsudio缺少KaiTi所解决的字体.zip
2020-03-24
Sort_1000pics机器学习小型数据集
2018-09-19
利用Atmega8控制NRF24L01进行无线远程遥控
2018-09-18
OpenGL高等计算机图形学(旋转风车,太阳系)源码及工程
2018-09-15
hadoop-3.0.1下载(hadoop-3)
2018-09-15
用51单片机与LCD12864实现俄罗斯方块游戏
2018-09-15
protues中51单片机程序和LCD液晶构成的计算器
2018-09-14
windows下python_opencv提取文件
2018-08-17
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人