草庐IT

Yolov5_DeepSort_Pytorch

全部标签

CUDA和显卡驱动以及pytorch版本的对应关系

1支持CUDA的GPU  支持CUDA的NVIDIAQuadro和NVIDIARTXCUDAGPU|NVIDIADeveloper您的GPU计算能力您是否正在寻找GPU的计算能力然后查看以下表格。您可以在这里了解更多计算能力。NVIDIAGPU为全球数百万台台式机笔记本电脑工作站和超级计算机提供动力加速了消费者专业人士科学家和研究人员的计算密集型任务开始使用CUDA和GPU计算并免费加入我们的NVIDIA开发者计划。https://developer.nvidia.com/zh-cn/cuda-gpus#compute 2怎么知道nvidia显卡该用什么CUDA版本呢?在官网查询nvidia显

Pytorch中RandomResizedCrop()的参数及用法

最近在看一些代码时,发现以前学习的函数有些遗忘了,在此复习一下,也希望能给大家带来一点有用的知识。这个RandomResizedCrop()函数跟ToTensor()这类函数一样,在torchvision中的transforms包里面。一般来说,用于图片的preprocess和dataaugmentation。importtorchvisiontrans=torchvision.transforms.RandomResizedCrop((224,224),scale=(0.8,1.0),ratio=(1.0,1.0))RandomResizeCrop()这个函数,最常用的三个参数是size,s

Jetson nano部署Yolov5目标检测 + Tensor RT加速(超级详细版)

    一、准备工具  二、烧录        三、搭配环境        四、试跑Yolov5        五、tensorRT部署yolov5前言:在工作或学习中我们需要进行部署,下面这篇文章是我亲自部署jetsonnano之后做出的总结,包括自己遇到一些报错和踩坑,希望对你们有所帮助: )一、准备工具读卡器SD卡 小螺丝刀网线(更改语言需要网络)二、烧录烧录镜像就是要把SD卡里的东西给完全清除,好比我们电脑重装系统一样,把SD卡格式化。 插上读卡器后会自动识别U盘,我的电脑会识别很多,弹出很多个U盘选项,这个是正常现象,只格式化一个就可以了。1.在本地的电脑上下载烧录的镜像,可以去官网

windows使用YOLOv8训练自己的模型(0基础保姆级教学)

目录前言一、使用labelimg制作数据集1.1、下载labelimg1.2、安装库并启动labelimg1.4、制作YOLO数据集二、使用YOLOv8训练模型2.1、下载库——ultralytics(记得换源)2.2、数据模板下载2.3、开始训练1、启动train.py,进行训练2、我们可以直接使用命令进行训练三、其他问题1、训练时,为什么会生成labels.cache文件?2、YOLOv8的训练图像和相应的标注文件应该放同一个文件夹里吗?还是应该分开放?3、yolov8训练自己的模型,大概需要多少图片作为训练集?效果合适?又需要训练多少轮呢?4、yolov8训练时会十分浪费电吗四、推荐视频

windows使用YOLOv8训练自己的模型(0基础保姆级教学)

目录前言一、使用labelimg制作数据集1.1、下载labelimg1.2、安装库并启动labelimg1.4、制作YOLO数据集二、使用YOLOv8训练模型2.1、下载库——ultralytics(记得换源)2.2、数据模板下载2.3、开始训练1、启动train.py,进行训练2、我们可以直接使用命令进行训练三、其他问题1、训练时,为什么会生成labels.cache文件?2、YOLOv8的训练图像和相应的标注文件应该放同一个文件夹里吗?还是应该分开放?3、yolov8训练自己的模型,大概需要多少图片作为训练集?效果合适?又需要训练多少轮呢?4、yolov8训练时会十分浪费电吗四、推荐视频

在ubuntu20.04上利用tensorrt部署yolov5(C++和Python接口)

在ubuntu20.04上利用tensorrt部署yolov5(C++和Python接口)‘下个博客是yolov7的部署’一、CUDA、CUDNN、TensorRT以及OpenCV安装CUDA安装CUDNN安装TensorRT安装OpenCV安装二、YOLOv5部署文件准备模型文件转换3.生成wts文件4.生成部署引擎5.端侧部署模型测试图片6.视频检测7.安卓部署8.C++部署一、CUDA、CUDNN、TensorRT以及OpenCV安装1、CUDA安装#CUDA=10.2#选择生成软链接,不需要安装驱动sudoshcuda_10.2.89_440.33.01_linux.run#查看CUD

在ubuntu20.04上利用tensorrt部署yolov5(C++和Python接口)

在ubuntu20.04上利用tensorrt部署yolov5(C++和Python接口)‘下个博客是yolov7的部署’一、CUDA、CUDNN、TensorRT以及OpenCV安装CUDA安装CUDNN安装TensorRT安装OpenCV安装二、YOLOv5部署文件准备模型文件转换3.生成wts文件4.生成部署引擎5.端侧部署模型测试图片6.视频检测7.安卓部署8.C++部署一、CUDA、CUDNN、TensorRT以及OpenCV安装1、CUDA安装#CUDA=10.2#选择生成软链接,不需要安装驱动sudoshcuda_10.2.89_440.33.01_linux.run#查看CUD

Pytorch自动求导机制详解

目录1.自动求导1.1梯度计算1.1.1 一阶导数 1.1.2二阶导数 1.1.3向量 1.2线性回归实战1.自动求导在深度学习中,我们通常需要训练一个模型来最小化损失函数。这个过程可以通过梯度下降等优化算法来实现。梯度是函数在某一点上的变化率,可以告诉我们如何调整模型的参数以使损失函数最小化。自动求导是一种计算梯度的技术,它允许我们在定义模型时不需要手动推导梯度计算公式。PyTorch提供了自动求导的功能,使得梯度的计算变得非常简单和高效。PyTorch是动态图,即计算图的搭建和运算是同时的,随时可以输出结果。在pytorch的计算图里只有两种元素:数据(tensor)和运算(operati

pytorch:参数pin_memory=True和non_blocking=True的作用

目录一、pin_memory二、non_blocking一、pin_memorypin_memory是dataloader()的参数,默认值为False,其作用是是否把把数据存放在锁页内存中。主机的内存根据物理内存(内存条)与虚拟内存(硬盘)进行数据交换分为锁页内存和不锁页内存:锁页内存:数据存放在物理内存上(内存条)上;不锁页内存:当物理内存(内存条)满载时,把部分数据转换到虚拟内存上(硬盘)上。锁页内存(pin_memory)能够保持与GPU进行高速传输,在训练时加快数据的读取,从而加快训练速度。因此,如果主机/服务器的内存足够大,建议把pin_memory设为True,如:trainlo

yolov8 瑞芯微 RKNN 的 C++部署

  上一篇博客yolov8瑞芯微RKNN和地平线Horizon芯片仿真测试部署写了在rknn模型的转换与PC端仿真测试,有网友希望写一篇在板子上部署的博文和开源提供C++代码。这一篇基于rknn板子进行C++部署,并开源提供完整的源代码和模型,供网友自行进行测试验证。特别说明:如有侵权告知删除,谢谢。【完整代码】代码和模型1、rknn模型准备  onnx转rknn模型这一步就不再赘述,请参考上一篇”yolov8瑞芯微RKNN和地平线Horizon芯片仿真测试部署“。上一篇提供了完整的模型和代码,如果仅仅是想验证模型,可以直接拿提供的rknn模型进行后续的步骤,本篇也是基于上一篇转好的rknn模