- 博客(23)
- 收藏
- 关注
RSS订阅原创 cuda两个向量相加
4. **CUDA 核心函数**：CUDA 提供了一些核心函数，如 `cudaMalloc`（分配内存）、`cudaMemcpy`（内存传输）、`cudaFree`（释放内存）等，这些函数对于管理 GPU 内存非常重要。3. **CUDA 内存管理**：CUDA 提供了不同类型的内存，如全局内存、共享内存和常量内存。CUDA 提供了原子操作和同步函数来实现线程同步。7. **错误处理**：学习如何处理 CUDA 函数返回的错误代码是很重要的，这可以帮助您调试和优化 CUDA 程序。
2024-03-05 19:45:45
244
原创 tensorrt 推理Unet网络（python && C++）
分类数量改为2，训练部分参考源码 ，源码中可能由于pytorch版本不一样需要改一下。我的输入是（1，3，720，1280），输出是（1，2，720，1280），源码。重要推理部分如下，C++部分改天上传。
2024-01-11 17:33:51
637
2
原创 C++ 不同线程之间类与类之间传递参数
在上述代码中，使用了一个互斥锁 mutex 来保护 dataQueue 的访问。循环接收数据的线程将接收到的数据放入队列，而其他线程则从队列中取出数据进行处理。通过互斥锁的加锁和解锁操作，确保了对共享数据的安全访问。（比如socket 接收到的数据，放入dataQueue中，然后在其他线程调用）
2023-12-05 13:47:04
107
1
原创 C++设计模式之创建型模式
1、单例模式（Singleton Pattern）：保证一个类只有一个实例，并提供一个访问它的全局访问点。例如，在一个多线程程序中，使用单例模式可以确保只有一个线程访问某个资源。C++设计模式的创建型模式可以分为单例模型、简单工厂模式、工厂方法模式、抽象工厂模式、原型模式、建造者模式。
2023-11-20 14:09:13
121
1
原创 tensorrt推理 onxx转engine代码（python），cyclegan网络推理（python、C++）
以cyclegan网络，分别用python 、C++对网络进行推理。将onnx文件导出为engine，FP16格式。C++版本首先配置cmakelists。
2023-11-03 19:48:48
1040
原创 onnx转wts文件与转engine文件的区别
Engine文件是TensorRT的序列化模型，其中包含神经网络的结构和权重。可以在TensorRT中使用Engine文件来进行推理。- WTS文件是权重文件，其中包含神经网络的所有参数，但不包括网络结构。可以在TensorRT中使用WTS文件来加载预先训练好的权重。因此，如果您只需要加载预先训练好的权重，则可以将ONNX转换为WTS文件。如果您需要进行推理，则应将ONNX转换为Engine文件。ONNX转换为WTS文件和转换为Engine文件有一些区别。
2023-09-22 09:45:25
666
原创 数据结构之排序算法
快速排序是一种常用的排序算法，它采用分治法策略，将原问题分解成若干个规模较小但类似原问题的子问题，递归地求解这些子问题，然后将这些子问题的解组合成原问题的解。选择排序是一种简单直观的排序算法，它的工作原理是每一次从待排序的数据元素中选出最小（或最大）的一个元素，存放在序列的起始位置，直到全部待排序的数据元素排完。冒泡排序是一种简单的排序算法，它重复地走访过要排序的数列，每次比较相邻的两个元素，如果它们的顺序错误就交换它们的位置。常见的排序算法有冒泡排序、选择排序、插入排序、快速排序、归并排序等。
2023-09-11 19:45:06
33
原创 使用QT多线程UDP发送结构体。
客户端接收代码，将客户端的一个类放入子线程。如需增加线程，接收其他端口的数据，可新实例化一个新的MyUDP类，并且修改端口号即可；首先是服务端，在构造函数中实例化UDP，并且创建成员函数进行发送，通过QTIMER类以10HZ的频率发送；
2023-09-05 18:01:32
970
原创 回调函数 C语言
上图的回调函数中打印输入的值；使用printNumbers调用回调函数，依次打印数组中的值。上图的回调函数有返回值，以实现累加的功能。
2023-08-25 10:04:22
55
1
原创 加载自己的数据集训练，以VGG和LeNet为例，自己做的笔记，避免忘记
将训练好的权重导出为onnx，导出的大小是通过加载数据集训练时的transform = transforms.Compose函数确定的。数据集为imagenet的格式，分为训练和验证两个文件夹，训练文件的子文件夹的名字代表具体的类别。
2023-08-09 17:18:21
192
1
原创 指针常量和常量指针的区别
指针常量是指一个指针变量被声明为常量，即指针变量的值（指向的内存地址）不能被修改，*代码指针标志，记法*在const前叫做指针常量 ，*在const后叫做常量指针。常量指针是指一个指针变量所指向的内存地址是常量，即指针变量所指向的内存地址不能被修改。
2023-07-21 18:20:57
38
1
原创 tensorrt 推理mnist数据集（python 版）
1、利用官方自带的trtexec.exe文件将Tensorrt自带的onnx导出为trt文件。2、导入相关的库，trt目录下有对应的tensorrt包。torch安装GPU版。
2023-07-13 18:01:58
108
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人