June vinvin
simple but not simple
展开
-
YOLOv5-6.1添加注意力机制(SE、CBAM、ECA、CA)
0. 添加方法主要步骤:(1)在models/common.py中注册注意力模块(2)在models/yolo.py中的parse_model函数中添加注意力模块(3)修改配置文件yolov5s.yaml(4)运行yolo.py进行验证各个注意力机制模块的添加方法类似,各注意力模块的修改参照SE。完整代码链接:1. SE(1)在models/common.py中注册SE模块class SE(nn.Module): def __init__(self, c1, c2, ratio=原创 2022-05-06 21:49:04 · 43870 阅读 · 118 评论 -
AlphaPose环境配置与测试
1. AlphaPose环境配置项目地址:https://github.com/MVIG-SJTU/AlphaPose(1)安装anaconda(参考链接1.1)https://blog.csdn.net/weixin_50008473/article/details/115250986?spm=1001.2014.3001.5501(2)创建虚拟环境:conda create -n alphapose python=3.6 -y(3)进入虚拟环境:conda activate alphapose原创 2022-04-07 14:53:27 · 5814 阅读 · 0 评论 -
Ubuntu 18.04 TensorRT 加速YOLOv5
1. 安装OpenCVgit获取OpenCV源代码:git clone https://github.com/opencv/opencv.git安装依赖库sudo apt-get install -y cmake#借助系统自带的opencv简化依赖安装步骤sudo apt-get install -y libopencv-dev编译源代码cd opencvmkdir buildcd buildcmake ..make -j 10替换已经安装的版本sudo make inst原创 2021-05-21 17:01:23 · 1156 阅读 · 7 评论 -
YOLOv5_DeepSORT_Pytorch训练自己的多目标跟踪模型
1 准备环境配置:https://blog.csdn.net/weixin_50008473/article/details/115250986?spm=1001.2014.3001.5501YOLOv5_DeepSORT_Pytorch代码地址:https://github.com/mikel-brostrom/Yolov5_DeepSort_PytorchDeepSORT论文:https://arxiv.org/pdf/1703.07402.pdf2 训练目标检测模型训练自己的目标检测模原创 2022-01-06 21:34:16 · 11455 阅读 · 84 评论 -
YOLOv5训练自定义的烟火数据集和COCO2017数据集
YOLOv5训练烟火数据集小记1、准备数据集(1)数据集具体结构内容如下:yolov5/data/images存放训练的jpg图片yolov5/data/annotations存放对应图片标注的xml文件yolov5/data/ImageSets新建Main文件夹存放划分的训练集、测试集等文件(2)运行voc2yolo5.py生成划分的训练集、测试集等文件import osimport random xmlfilepath=r'annotations'saveBasePath=r"I原创 2021-03-30 17:52:39 · 6328 阅读 · 8 评论 -
YOLOv4训练自定义数据集
目录1.环境配置2.权重下载3.数据集处理4.修改配置文件4.1 修改data/voc.names4.2 修改data/voc.data4.3 修改cfg/yolov4.cfg4.4 修改Makefile文件5.开始训练文献地址:https://arxiv.org/pdf/2004.10934.pdf源码地址:https://github.com/AlexeyAB/darknet1.环境配置https://blog.csdn.net/weixin_50008473/article/details/1原创 2022-03-18 14:06:39 · 4327 阅读 · 0 评论 -
YOLOv3训练自定义数据集
1. 下载YOLOv3和权重文件YOLOv3:git clone https://github.com/pjreddie/darknetyolov3.weights:wget https://pjreddie.com/media/files/yolov3.weightsdarknet53.conv.74:wget https://pjreddie.com/media/files/darknet53.conv.742. 数据集处理(1)在scripts文件夹下按如下目录创建VOCdevkit 文件夹原创 2022-03-28 21:12:01 · 4041 阅读 · 0 评论 -
数据集处理
1.筛选voc数据集中指定类别以猫狗为例:# -*- coding: utf-8 -*-import osimport shutilann_filepath = r'C:/Users/double_ww/Desktop/Annotations/' # 数据集标注文件夹路径img_filepath = r'C:/Users/double_ww/Desktop/JPEGImages/' # 数据集图片文件夹路径img_savepath = r'C:/Users/double_ww/Desk原创 2021-07-14 10:33:12 · 446 阅读 · 0 评论 -
目标检测数据集标注工具之labelimg的使用
1.安装:若电脑安装了python3,cmd命令行输入以下命令pip install labelimg2.打开:终端输入labelimg3.使用:(1) 设置自动保存(2)工具栏使用Open Dir:需要标注的图片文件夹Change Save Dir:标注文件存放的文件夹Next Image:切换下一张,对应快捷键dPrev Image:切换上一张,对应快捷键a(3)标注Create\nRectBox(对应快捷键w):框选目标,文本框标注类别即可标注完后得到标注的xml文件,内原创 2021-08-03 09:08:07 · 968 阅读 · 1 评论 -
docker部署ubuntu18.04深度学习环境——cuda11.1、cudnn8.0.5、pytorch1.8.0、tensorrt7.2.2.3、opencv4.4.0
这里写目录标题一级目录二级目录三级目录1 安装nvidia显卡驱动2 安装docker3 安装docker compose4 安装nvidia-docker5 下载 nvidia CUDA 镜像6 安装python3.8环境7 安装pytorch1.88 提交容器成为新镜像常用命令附录一级目录二级目录三级目录1 安装nvidia显卡驱动参考2.2更新驱动:https://blog.csdn.net/weixin_50008473/article/details/1152509862 安装dock原创 2021-08-07 00:30:41 · 7803 阅读 · 15 评论 -
YOLOv5—RTX3080 conda环境配置cuda11.1、cudnn8.0.5、pytorch1.8.0、tensorrt7.2.2.3、opencv4.4.0、deepstream5.1.0
YOLOv5—RTX 3080 conda环境配置pytorch1.X86版本的Anaconda的安装1.1 安装anaconda(1)下载页面链接:https://www.anaconda.com/products/individual根据系统信息选择相应的安装包,如下图所示:(2)安装步骤1)cd 文件下载目录2)执行命令:bash Anaconda3-2020.11-Linux-x86_64.sh安装过程中:<1> 阅读注册信息,回车继续<2> 同意注册原创 2021-03-26 18:12:24 · 4475 阅读 · 6 评论 -
nvidia jetson xavier刷机以及pytorch1.7.0环境配置
1. nvidia jetson xavier刷机准备:安装有Ubuntu 1804系统的PC一台、Jetson设备和电源、Type-C接口的USB连接线一根刷机步骤:1.使用Type-C接口的USB线把PC与Jetson设备连接起来(设备端口为Type-C的OTG端口)2.在Jetson关机状态下,使用细针按住recovery button不放,再按开机键,进入recovery模式 (在机器外壳侧面标有rc的小孔内,即是recovery button)3.解压文件:tar xzvf mfi_nx原创 2021-07-07 13:43:04 · 1030 阅读 · 0 评论 -
NVIDIA AGX jetpack4.3版本更新至jetpack4.4(包含CUDA、TensorRT环境)
1 硬盘挂载(1)显示硬盘和所属分区情况,执行命令:sudo fdisk -l如下图所示:此时240G的硬盘没有识别出来。(2)对硬盘进行分区并格式化,执行如下命令1>sudo parted /dev/nvme0n12>mklabel gpt3>mkpart primary 2048s 100%4>quit5>sudo mkfs.ext4 /dev/nvme0n1p1(3)查看操作后的硬盘详情,执行命令:sudo fdisk -l如上图,/dev/nvme原创 2021-09-04 15:41:55 · 1589 阅读 · 1 评论 -
Ubuntu18.04多版本CUDA和CUDNN的切换
本篇为CUDA11.1和CUDA11.2的切换。前言(若已安装,跳过该部分)1. 安装cuda11.2(1)wget https://developer.download.nvidia.com/compute/cuda/11.2.2/local_installers/cuda_11.2.2_460.32.03_linux.run(2)sudo sh cuda_11.2.2_460.32.03_linux.run设置环境变量1>sudo vim ~/.bashrc2>添加以下两行内容原创 2021-04-01 17:48:54 · 977 阅读 · 0 评论 -
Ubuntu18.04挂载2T机械硬盘
Ubuntu18.04挂载2T机械硬盘1.显示硬盘和所属分区情况执行命令:sudo fdisk -l此时2T的机械硬盘没有识别出来。2. 对硬盘进行分区,分区/dev/sda,挂载该块硬盘执行命令:sudo fdisk /dev/sdahp@hp:~$ sudo parted /dev/sdaGNU Parted 3.2Using /dev/sdaWelcome to GNU Parted! Type ‘help’ to view a list of commands.(parted原创 2021-03-31 14:28:06 · 1054 阅读 · 0 评论 -
安装Ubnutu18.04图形化界面
ubuntu无法进入桌面,安装ubuntu-desktop失败1. 解决方法Ctrl+Alt+F1(或者F2-F8)进入命令行界面,输入账户名和密码登录卸载Xwindows软件 sudo apt-get autoremove ubuntu-desktop 安装所有依赖软件 sudo apt-get update sudo apt-get install ubuntu-desktop重启 sudo rebootP.若执行sudo apt-ge原创 2021-04-21 19:11:53 · 855 阅读 · 2 评论 -
安装ubuntu18.04操作系统
本次操作使用光盘安装。将引导光盘插入服务器光驱中,重启电脑按F12进入启动盘选择第一项IPV4,如下图:出现下图界面,需等待两分钟:进入新界面如下,选择F9:选择第二项:一直默认安装,!!!:Installation type选择Erase disk and install UbuntuInstall Now完成安装。重启。若出现如下问题:参考: https://blog.csdn.net/weixin_50008473/article/details/116458805?原创 2021-05-07 09:28:30 · 716 阅读 · 0 评论