- 博客(33)
- 收藏
- 关注
RSS订阅原创 U2Net、U2NetP分割模型训练---自定义dataset、训练代码训练自己的数据集
前言博客很久没有更新了，今天就来更新一篇博客吧，哈哈；最近在做图像分割相关的任务，因此，写这么一篇博客来简单实现一下分割是怎么做的，内容简单，枯燥，需要耐心看，哈哈；博客的内容相对简单，比较适合刚接触分割的同学参考学习（这篇博客在算法训练上没有涉及到训练策略、数据增强方法，特意留下余地处给大家自行发挥）内容简介U2Net算法介绍本博客训练效果截图展示本博客代码框架介绍数据集数据集准备自定义datasetu2net、u2netp网络结构定义训练代码模型推理代码总结以及博客代码的
2022-04-20 23:30:36
12406
35
原创 python使用Pillow/Opencv对图片进行等比缩放
记录一下小脚本吧，很久没写博客了。。代码用来对照片进行等比缩放；原图（下载自百度图库）代码# coding: utf-8# author: hxy# 2021-8-5"""1. 图片的等比缩放;（m,n） <- (scale_size, scale_size)2. 缩放后的图片放置在中间区域，其它区域像素填0；"""import cv2from PIL import Imageimport numpy as npclass UniformScale: d
2021-08-06 14:20:26
1272
原创 YoloV3-Openvino推理引擎在Tiger Lake上进行benchmark测试
博客内容章节简介：软件环境和硬件设备的简要介绍使用openvino自带的downloader模块进行模型的下载、转换和量化；模型转换： pb模型转换至openvino支持的xml和bin文件；模型量化：生成FP32和FP16的模型文件；使用openvino量化工具POT进行模型量化FP32 model -> INT8 modelFP16 model -> INT8 model测试量化好的模型在CPU、GPU以及CPU+GPU硬件条件下的推理性能；source
2021-06-01 17:18:58
1176
1
原创 2021年中科院先导杯计算应用大奖赛-智能交通检测之图像数据增强（方法一）
前言：由于本人最近在相关公众号中看到了这个比赛（如下图），因此，就报名参与了该比赛。参与比赛的开始阶段，拿到数据集后，想到了一个数据增强的办法，感觉应该挺不错的。因此，在本博客中写出，供大家参考使用，希望对大家有所帮助，共同学习、进步！ps：本人也在疯狂“压榨”GPU中！接下来，进入正题一. 原始数据集展示；二. 数据增强操作(如下图)；1. 数据增强原始图片和增强图片的对比；2. 数据增强的代码；3. 验证数据增强后标签的准确性（可视化标注框信息）；一、原始数据集展示原始
2021-04-26 22:48:14
1149
9
原创 NVIDIA-TLT训练行人检测模型（一）----算法模型的训练(finetuning)
前言在博客阅读前需要说明，本博文为系列文章，通过阅读文章，您将会学习到如下内容：使用NVIDIA Transfer Learning Toolkit 工具训练(finetuning)出一个行人检测算法模型；将训练好的算法模型部署到NVIDIA Jetson TX2上，结合deepstream sdk实现模型对多路rtsp视频流进行实时的行人检测；思路的简介（目录）在我们做一件事之前，需要思考；做事的思路，做事的方式，这样我们才能顺利、高效的完成这件事；Docker-ce的安装Nvidia
2020-10-10 14:44:21
2488
3
原创 阿里天池-太阳黑子耀斑分类（赛道一）之数据处理脚本
前言前段时间，在微信朋友圈内看到有朋友在打比赛，然后，我就抱着尝试的心态去参加了一把；一个人成队，就报名参加了。比赛截止到7月26号，本人被其他参赛队伍吊打！哈哈哈哈！这篇文章主要是分享一下本人参赛时用于数据预处理的脚本，自己记录一下，也供大家学习参考！[本人被各位大佬吊打之后，最终的成绩如下图，由于工作比较忙(zijilan)，就只提交了两次成绩]比赛中，数据处理的步骤数据共计三大类，alpah、beta、betax，并且原始数据量不算很多，因此，首先想到的就是进行数据增强操作；在三大类别
2020-08-06 10:32:13
818
2
原创 Yolo txt标签文件转换至Kitti txt标签格式
由于最近用的算法要用到kitti数据格式的标签文件，但是手头的数据呢都是yolo的txt标签文件，于是写了一个脚本来转一下标签文件；（目标检测使用）ps：该脚本只是针对一类目标进行标签文件的转换，并且在图片内只有一个被标注的目标。若您的类别数量大于1，请在代码的基础上进行简要修改；脚本一：用于检查是否现有标签文件有空的，有空的标签文件的话，就将对应的标签文件以及相应照片删除；check_labels.py# coding: utf-8# author: hxy# 2020-6-15"
2020-06-16 11:22:53
787
原创 yolov3_tiny.onnx转trt采用tensorrt加速模型推理
既然上一篇博客都把yolov3-tiny.weights转onnx做了，推理也测了。那么呢，就再直接转个trt模型吧。这样感觉博客的内容就更加连贯了吧，实用性貌似会更加强吧。（如果没看过yolov3-tiny转onnx这篇博客的，请点这，带你飞过去）这篇博客的内容是接着上一篇博客写的，所以呢，这里就直接进入主题，上代码！！本文的目录：onnx模型转trt文件（yolov3-tiny.onnx）；实用tensorrt进行推理；测试推理效果（视频文件）；首先呢，你得要安装好tensorrt，至
2020-05-26 22:04:36
4471
10
原创 yolov3-tiny原始weights模型转onnx模型并进行推理
时隔不知道多少天，我记起来我还有部分博客没写完(偷懒)，所以不能偷懒把它完成！！ 这篇博客的主要内容将yolov3-tiny.weights模型转换到.onnx模型；使用onnnxruntime-gpu（加速）模型推理；（不知道这里用加速形容是否合适）照片推理测试；视频文件推理测试；（注：本文用的模型为本人日常测试的时候训练出来的一个小模型，能够检测车辆、行人、人脸，但是效果有限，仅限用于本文的博客内容测试使用）接下来，我们就按照博客内容一步一步的来进行我们的学习（ctrl+c、ctrl+v）
2020-05-26 11:23:28
5118
13
原创 Visdrone2019数据集.txt标签文件转换为voc格式.XML标签文件
最近有同学问是否有Visdrone数据集的xml文件，由于本人之前训练数据的时候没有保存xml文件，所以无法共享。为了解决这个问题，重新写了转换代码并贴出，供大家共同学习使用。（文末附上数据下载网盘地址）# coding: utf-8# author: HXY# 2020-4-17"""该脚本用于visdrone数据处理；将annatations文件夹中的txt标签文件转换为XML...
2020-04-17 15:04:47
6448
17
原创 yolov3模型部署实战weights转onnx并推理
前言：最近比较忙(懒)，本学渣在写毕业论文(好难受)，所以博客的更新不是那么有效率，哈哈；本文的目的是用实际的代码带你一步一步的实现darknet模型框架的部署流程；（当然darknet算法的训练在本人之前的博客也有写，你要是串起来，那也是很棒的！）实现yolov3/yolov3-tiny模型从.weights模型转换到.onnx模型然后再转换到.trt模型；当然，本文也是本人自己对自己学...
2019-12-26 16:44:09
10759
21
原创 Darknet YoloV3调用python接口进行批量图片检测
本代码主要用于调用darknet的python接口，进行图片的批量检测，并且将检测结果的图片保存到指定目录话不多说，直接上代码，将代码复制并且保存到darknet的python目录下执行。batch_img_detect.py# coding: utf-8# author: HXY"""对照片进行批量检测；并将检测结果照片存储；"""from ctypes import *im...
2019-11-28 16:26:21
2175
5
原创 Ubuntu16.04源码交叉编译opencv-3.4.6与libjpeg-turbo
安装流程安装nasm源码编译安装libjpeg-turbo-2.0.1源码交叉编译安装opencv3.4.6一. 安装nasm1.终端中指令安装(推荐)sudo apt install nasm2.或者源码编译安装https://www.nasm.us/pub/nasm/releasebuilds/2.14.02/二. 源码编译安装libjpeg-turbo-2.0.1...
2019-11-26 14:10:44
741
1
原创 CUDA版本和NVIDA显卡驱动版本的匹配信息
很多时候在搭建深度学习环境的时候会纠结如何选择显卡驱动版本和CUDA版本问题，所以，为了更加方便的找到相关信息，从NVIDIA官网截取下来了这个配置表，希望对大家有所帮助！！官网链接：https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html...
2019-11-14 17:03:16
1570
原创 人脸对齐--采用dlib库的68_face_landmark进行人脸对齐操作
简单说说人脸对齐操作的部分作用人脸对齐操作的目的就是能够把检测到的水平角度不正的人脸采用数学的方式进行角度的纠正。从而，在一定程度上提升后期人脸识别的精确度。人脸对齐操作的基本步骤人脸检测人脸关键点信息检测(眼睛，鼻子，嘴巴，下吧等…)人脸对齐人脸对齐的方法有很多，本文只是采用dlib库提供的68点关键点信息检测的模型来实现人脸对齐操作，本人能力和技术有限，代码和思路供大家参考和学...
2019-11-06 10:06:05
2347
9
原创 python随机从批量文件夹中筛选一定数量的照片/文件夹批量重命名
日常使用的一些小脚本，码一下！作用：随机从若干的文件夹内筛选一定数量的照片，并且以相同的文件名保存到新的目录中random_select_pic.py#-*-coding: utf-8-*-#author: hxy'''从每个文件夹中随机选择nums张照片，保存到新的文件目录，用来制作测试数据集'''import osimport shutilimport timedef...
2019-10-10 14:30:47
1167
3
原创 MTCNN+Face_recognition实时人脸识别(二)训练自己的数据/多进程实时视频人脸识别
文章内容介绍本文在上一篇博客的基础上，进行人脸识别特征模型的训练，以及将训练好的模型应用于图片上人脸识别和实时视频中的人脸识别。本文内容中代码会偏多，若是代码中有误，或者不优化，欢迎指出问题，留言，交流，谢谢！话不多说，直接上code！！！code跑起来看到实际的效果胜过千言万语！！目录1.人脸识别数据集的准备2.照片中的人脸识别以及底库人脸特征模型的生成3.实时视频流人脸识别以及底库...
2019-10-07 21:20:37
4057
11
原创 MTCNN+Face_recognition实时人脸识别（一）基础环境的搭建/算法结合
文章内容介绍这段时间一直有点小懒，写文章的进度有点慢。。无奈。。。本文的目的是将mtcnn人脸检测算法和face-recognition算法结合起来，实现实时视频中人脸识别（1:N）。本project将会以两篇博客的形式书写，project的代码将会等文章书写完同步到本人的github上，有兴趣的朋友记得给个小星星。注意：由于本人能力有限，本博客不对算法做解读，只是从代码层面（搭积木）实现了...
2019-09-24 17:38:57
2929
1
原创 使用face-recognition时遇到的报错： CUDNN_STATUS_EXECUTION_FAILED
最近利用face recognition库做相关任务的时候遇到了报错，一个小问题，我却花了三天才解决。这个过程简直无奈，无奈，无奈…具体的错误信息如下:RuntimeError: Error while calling cudnnConvolutionForward( context(), &alpha, descriptor(data), data.device(), (const...
2019-09-04 14:43:56
2685
3
原创 MTCNN + Deep_Sort实现多目标人脸跟踪之Deep_Sort算法部分(二)
前言：本文的测试思路仅供参考和学习，希望能和大家分享、交流相关的学习经验！同时，本人的文字功底不是那么好，所以就直接上代码，请多多谅解！文章内也没对算法做解读，所以，若是想要了解算法，可以参考相关算法的解读博客！！文章最后贴上Project的github链接 ～目录一. MTCNN实时视频人脸检测二. Deep_Sort与MTCNN结合三. 结果截图展示一. MTCNN实时视频人...
2019-08-22 11:20:22
4059
16
原创 MTCNN + Deep_Sort实现多目标人脸跟踪之MTCNN人脸检测部分(一)
最近在做一个人脸识别的任务，所以，就玩了一下mtcnn！！将部分内容写成了博客的形式分享学习！博客内容较为简单，但是，能够实现需求。不足之处多多谅解！！谢谢～目录MTCNN TensorFlow版人脸检测算法实现流程算法环境的搭建数据集的准备MTCNN检测模型的训练人脸检测模型的效果测试以及视频检测代码的修改视频人脸检测demo算法环境的搭建实验机器配置为： intel i7...
2019-08-09 15:17:08
2486
1
原创 python-opencv间隔拍照并保存照片
日常用到的小脚本，用opencv调用笔记本摄像头进行间断式的拍照，并且保存照片#coding: utf-8-*- #Author:lxz-hxy'''opencv每隔若干秒拍照并且保存'''import osimport timeimport threadingfrom cv2 import cv2 as cv2 #在vscode里面直接import cv2会显示代码有问题，其实...
2019-07-30 16:31:15
10037
1
原创 Tensorflow GPU版本与CUDA、Cudnn以及相关库的构建配置表
由于最换了新机器，需要安装新环境和新版本的tf-gpu，总是遇到一些乱七八糟的问题。后面才发现，很多问题都是相关的库与tf版本不匹配而导致的。因此，为了快速解决问题，就从官网将其在Linux下有效构建的配置表摘了下来，供参考、学习、使用；TF官网地址：https://tensorflow.google.cn/install/sourcelinux系统下有效的构建配置表：...
2019-07-22 11:10:54
406
1
原创 Keras Yolo_MobileNet训练自己的数据
前言：最近在做一个小的project，要用到小模型。所以，特此尝试了一下用Mobilenet进行相关的模型训练，以及将模型转换成pb格式。本文中，部分代码已做简单的修改，具体修改请看正文部分。 同时，本次模型训练只针对了一类目标进行检测，多类目标的训练请自行修改相关部分代码（其实只需要修改name.txt文件即可）。由于本人之前写过部分博客，为了精简本文的内容，部分步骤与之前的博客有重复，所以...
2019-06-21 21:19:56
3720
41
转载 Keras日常使用的一些小技巧（转载）
本文为转载的文章，在学习过程中发现作者写的方法很使用，因此将其转载，若在阅读过程中有问题，可看原文链接，若有侵权将及时删除！！原文地址：https://kexue.fm/archives/5765层的自定义这里介绍Keras中自定义层及其一些运用技巧，在这之中我们可以看到Keras层的精巧之处。**基本定义方法 **在Keras中，自定义层的最简单方法是通过Lambda层的方式：fro...
2019-06-10 19:39:43
555
1
原创 根据图片名字生成文件夹并归类+批量重命名照片名
最近在处理手头的部分图片数据，所以写了码了点code，也许质量不高，供大家参考，欢迎各路大神纠正问题！！1.该代码根据照片的名字来生成文件夹，并且将同类照片放置同一文件夹creat_dir.py#-*-coding: utf-8-*-#Author: AIBC-MrH#该代码为数据处理用，根据照片的名字来生成文件夹；#并且将source_dir目录下同一类别的照片保存至对应的文件夹内；...
2019-05-31 09:13:44
9960
9
原创 使用YOLOV3快速实现UAV视角的目标检测--VisDrone2019
目录：1. Visdrone2018数据集的介绍及下载；2. Visdrone2018数据集标签格式转换；3. YoloV3算法模型代码的调整以及模型的训练；4. 总结；前言：学渣的生活就是拿着各种数据集在算法上各种训练、各种玩……当然，个人觉得还是有意思的，在你有一台Bebop2无人机的情况下，结合本人前面的博客，搭建一个无人机实时目标检测平台，再配合这个检测模型，还是相对有点意思的...
2019-05-12 17:26:23
18459
157
原创 YoloV3_Keras模型MAP测试
目录测试图片的准备数据标签文件的格式转换(XML转换为txt格式)模型对测试图片的检测及检测结果坐标文件的生成MAP测试结果说明：1、本次试验中，map的测试采用的是人脸图片数据；2、本实验中模型的训练/MAP测试只使用了部分实验数据，因此，最后的MAP值并不高，仅供参考；3、关于如何如何训练人脸识别的模型可以参考本人上一篇博客，仅供互相学习、交流；一、 测试图片的准备...
2019-04-11 00:42:41
4133
37
原创 Keras_Yolov3 实现人脸检测
目录1. 实验环境的搭建2. 人脸数据集的介绍及准备3. 相关代码的修改及调整4. 算法模型的fine-tuning5. 模型检测结果一．实验环境的搭建本次实验所需要的环境配置如下：Python=3.5、tensorflow-gpu==1.4.0、Keras环境的安装：创建虚拟环境(需安装anaconda)conda create...
2019-04-02 14:13:43
4438
26
原创 YOLOV3—进行车辆检测
目录数据集的准备1.数据集的介绍及下载；2.数据集格式转换----转换为VOC数据格式；Yolov3相关代码的调整开始训练模型测试一． 数据集的准备1.数据集介绍：UA-DETRAC数据集主要拍摄于北京和天津的道路过街天桥，多为摄像头俯视视角；数据集：DETRAC-Train-Images (5.22GB, 60 sequences)DETRAC-Test-Images...
2019-03-20 11:41:55
22094
137
原创 ROS+Bebop2无人机+YOLO算法实现无人机视角的实时目标检测
前言：很久之前，用TK1玩过一段时间的ROS，再加上各种硬件(Arduino、激光雷达、编码电机等)，模仿着做过Turtlebot小车，实现了部分Turtlebot部分的功能，最后因为雷达被我玩烧了，所以，就玩完了。最近，身边有几台Bebop2无人机，然后又在玩深度学习的目标检测，因此，就尝试了一下将目标检测算法和无人机结合在一起玩一玩。话不多说，先看玩成功的视频链接：B站：https:/...
2019-03-05 10:25:26
12083
26
原创 Mask Rcnn(Keras)学习笔记
实验环境搭建Part1:实验准备实验项目相关的网址：https://github.com/matterport/Mask_RCNN （project网址）https://github.com/waleedka/coco （cocoAPI）实验所需要的数据介绍及实验数据的下载：实验数据为COCO2014train2014.zip，val2014.zip， test2014.zip5k...
2019-03-04 10:30:59
916
1
原创 深度学习环境搭建：Ubuntu16.04+NVIDIA-387+ CUDA-8.0+Cudnn-v5.1+Tensorflow-gpu 1.2+Opencv3.2+Caffe安装
安装流程：1. Ubuntu16.04系统安装；2. NVIDIA-387显卡安装；3. CUDA-8.0安装；4. Cudnn-v5.1安装；5. Tensorflow-gpu 1.2版安装；6. Opencv3.2安装；7. Matlab2016b安装；8. Caffe安装；一. Win10+ubuntu16.04双系统安装电脑型号：华硕飞行堡垒配置：i7-7700 ，8...
2019-03-02 21:36:57
1139
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人