- 博客(28)
- 资源 (5)
- 收藏
- 关注
原创 tensorflow2读取frozen_model.pb中的节点名字
import tensorflow as tffrom tensorflow import kerasimport os model_dir = '/Users/demo/Documents/models/test3/'model_name = 'frozen_graph.pb' def create_graph(): with tf.io.gfile.GFile(os.path.join(model_dir, model_name), 'rb') as f: grap
2021-09-19 00:56:09 488
原创 深度可分离卷积(参数量与计算量)VS普通卷积(参数量与计算量)
这里写自定义目录标题欢迎使用Markdown编辑器新的改变功能快捷键合理的创建标题,有助于目录的生成如何改变文本的样式插入链接与图片如何插入一段漂亮的代码片生成一个适合你的列表创建一个表格设定内容居中、居左、居右SmartyPants创建一个自定义列表如何创建一个注脚注释也是必不可少的KaTeX数学公式新的甘特图功能,丰富你的文章UML 图表FLowchart流程图导出与导入导出导入欢迎使用Markdown编辑器你好! 这是你第一次使用 Markdown编辑器 所展示的欢迎页。如果你想学习如何使用Ma.
2021-07-27 10:48:52 1711
原创 yolov5推理类
import argparseimport time, osfrom pathlib import Pathimport cv2import torchimport torch.backends.cudnn as cudnnfrom numpy import randomfrom models.experimental import attempt_loadfrom utils.datasets import *from utils.general import check_img_s
2021-07-15 22:17:32 621
原创 数据图片梯形矫正
import osimport cv2import numpy as npimage_path = '/home/hs/data/new_data/data'corrected_image_path = '/home/hs'lsPoints = []pointsCount = 0pointsMax = 4def correct_image(img, corrector_position, img_name): print('corrent image') image = im
2021-07-14 10:40:49 549
原创 小图检测数据处理删除边缘小框
import cv2import xml.etree.ElementTree as ETimport pdbimport os# path_ori="/home/huangs/00-Projects/FruitsRecognition/DataSet/MakingDataSet/100fruits/phoneShot4/img/"# xml_fold = '/home/huangs/00-Projects/FruitsRecognition/DataSet/MakingDataSet/100f
2021-07-13 16:28:45 99
原创 yoloTxtToXml.py
from os.path import isfile, isdir, joinimport random, yamlimport sys, cv2 ,osfrom shutil import copyfilefrom distutils.dir_util import copy_treefrom xml.dom import minidomimport xml.etree.ElementTree as ETfrom lxml.etree import Element, SubElement,.
2021-07-12 18:06:14 80
原创 docker-compose安装问题
1、版本问题,版本不匹配,解析不了docker-compose文件解决:sudo -i,重装docker-compose,先选好版本号,我选的是1.26.2版本的docker-compsoe.step1:which docker-compose/usr/bin/docker-composestep2:$sudo rm /usr/bin/docker-composestep3:curl -L https://github.com/docker/compose/releases/download
2021-07-12 12:03:31 644
原创 读取xml文件转移picture
import os.pathimport shutilimport cv2import xml.etree.ElementTree as ETimport pdbimport osimport mathdef each_file(imgPath, xmlPath, new_imgPath, new_xmlPath): ''' 读取每个文件夹,将遇到的指定文件统统转移到指定目录中 :param filepath: 想要获取的文件的目录 :param new_f
2021-07-09 11:07:07 435
原创 create_train_data_yolov5
from os import listdir, systemfrom os.path import isfile, isdir, joinimport random, yamlimport sys, cv2, osfrom shutil import copyfilefrom distutils.dir_util import copy_treefrom xml.dom import minidomonly_val = Falsewhere = 'sh1'if only_val:
2021-07-08 20:12:26 156
原创 批量修改图片name
#coding:utf-8from PIL import Imageimport osimport globimport xml.etree.ElementTree as ETimport xml.dom.minidomimgsdir = "/home/hs/yolov5/yolov5/data/imageyili"newimg = "/home/hs/yolov5/yolov5/data/imageyili"n =1if not os.path.exists(newimg):
2021-07-08 20:11:03 100
原创 change Xml Object name
import osimport os.pathfrom xml.etree.ElementTree import parse, Elementdef changeAll(xml_fold): ''' xml_fold: xml存放文件夹 new_name: 需要改成的正确的名字,在上个例子中就是cow ''' files = os.listdir(xml_fold) cnt = 0 for xmlFile in files: f
2021-07-08 20:06:13 115
原创 movedata
import os.pathimport shutilimport cv2import xml.etree.ElementTree as ETimport pdbimport osimport mathdef deletesinglefile(xmlpath,picpath, delclass,new_xmlpath,new_picpath): for annoName in os.listdir(xmlpath): try: # picF
2021-07-08 20:02:01 214
原创 letterbox
import cv2import xml.etree.ElementTree as ETimport pdbimport osimport math#图片拓展,使用letterbox方式填充,补黑填充#960*300----->960*960,补黑边填充#300*400----->400*400,def letterbox(old_img_path, new_img_path, old_xml_path, new_xml_path):#old_img_path new_img_
2021-07-08 20:00:46 266
原创 筛选问题图片数据
import os# from keras.preprocessing import imagefrom skimage import io# imgDir = "/home/hs/tools/rotation/YOLOv5_DOTA_OBB-master/dataliuqunwei/imgs"imgDir = "/home/hs/tools/rotation/images"txtDir = "/home/hs/tools/rotation/YOLOv5_DOTA_OBB-master/data
2021-07-08 19:59:30 104
原创 mosaic
from PIL import Image, ImageDrawimport numpy as npfrom matplotlib.colors import rgb_to_hsv, hsv_to_rgbimport mathimport xml.dom.minidomimport numpy as npimport sysimport cv2import globimport osimport xml.dom.minidomimport argparseimport random
2021-07-08 19:55:12 155
原创 大图裁剪小图同时裁剪annotation
import osimport sysimport argparseimport cv2 import numpy as npfrom PIL import Image, ImageDrawimport xml.dom.minidomfrom xml.etree.ElementTree import ElementTree,Element,parsefrom xml.dom import minidomimport xml.etree.ElementTree as ETimport x
2021-07-08 19:53:48 205
原创 python批量修改图片name和annotation的name
#coding:utf-8from PIL import Imageimport osimport globimport xml.etree.ElementTree as ETimport xml.dom.minidomimport randomdef changePicAndXmlName(oldImg, oldXml, newImg, newXml): n = 1 if not os.path.exists(newImg): os.mkdir(newImg
2021-07-08 19:51:46 134
原创 yolo格式转voc格式
下面展示一些 内联代码片。// 传入图片路径以及txt路径import argparsefrom os.path import isfile, isdir, joinimport random, yamlimport sys, cv2 ,osfrom shutil import copyfilefrom distutils.dir_util import copy_treefrom xml.dom import minidomimport xml.etree.ElementTree as
2021-07-08 18:17:29 203
转载 web服务器、WSGI跟Flask(等框架)之间的关系
最近在部署深度学习模型到web端,需要熟悉一些、apache、flask等知识,之前特别混乱对于web服务器、web框架这方面;业务场景就是:客户端从发送一个 HTTP 请求到 Flask 处理请求,HTTP 请求分别经过了 web服务器层,WSGI层,web框架层,这三个层次,最后返回这个HTTP 请求的响应。不同的层次其作用也不同,下面简要介绍各层的作用图1:web服务器,web框架与 WSGI 的三层关系Web服务器层加粗样式对于传统的客户端 - 服务器架构,其请求的处理过程是,客户端向服务器
2021-04-22 11:03:22 478
原创 pytorch训练 loss=inf或者训练过程中loss=Nan
造成 loss=inf的原因之一:data underflow最近在测试Giou的测试效果,在mobilenetssd上面测试Giou loss相对smoothl1的效果;改完后训练出现loss=inf原因: 在使用log函数时出现 data underflow解决方法:增加一个bias原代码 # match wh / prior wh g_wh = (matched[:, 2:] - matched[:, :2]) / priors[:, 2:] g_wh = torc
2021-03-30 20:50:55 7827
原创 Deformable Convolutional Networks(可变形卷积DCN)理解
Deformable Convolutional Networks(可变形卷积DCN)理解1、why???DCN视觉识别中的一个关键挑战是如何在目标尺度、姿态、视点以及部件变形中适应几何变化或建模几何变换。说白了就是:传统的卷积采用固定尺寸的卷积核,不能很好地适应几何形变,于是可以适应几何形变的它来了。2、what???DCN可变形卷积,卷积核的形状是可变的,也就是感受野可以变化,但注意感受野的元素是“不变”的。可变性卷积的流程为:1、原始FM(大小为h*w),记为U,经过一个普通卷
2021-03-26 14:27:47 836
原创 QT+ffmpeg+rtsp+多摄像头实时显示
QT+ffmpeg+rtsp+多摄像头实时显示直接上图,此工程是在业界某大神的一个demo上拓展的,可直接输入rtsp地址,支持四通道实时显示四个rtsp视频流,当然该工程拓展性强,可读性强,不是用定时器触发,采用信号槽机制实现视频流播放。代码简单,逻辑明了。在qt+ubuntu+ffmpeg+rtsp保证编译可用,当然也可以在Windows,windows下要将对应ffmpeg库文件替换.双击放大目标窗口双击第一个小窗...
2020-10-28 15:19:02 4979 180
原创 批量修改整理数据集
下面展示一些 内联代码片。批量 整理目标检测数据 批量整理用于路径下各个文件夹内部图片和XML标签文件的批量重命名路径部分需要小修改,主体部分可以直接套用talk is cheap,show code#coding:utf-8from PIL import Imageimport osimport globimport xml.etree.ElementTree as ETimport xml.dom.minidom# i = 1dir_data = "/home/hs/b
2020-09-04 09:42:42 292
原创 手撸ssd
// An highlighted blockimport tensorflow as tfimport numpy as npimport cv2class ssd(object): def __init__(self): self.feature_map_size = [[38, 38], [19, 19], [10, 10], [5, 5], [3, 3], [1, 1]] self.classes = ["aeroplane", "bicy..
2020-08-23 15:02:35 388
原创 手撸SSD
下面展示一些 内联代码片。// A code blockvar foo = 'bar';// An highlighted blockimport tensorflow as tfimport numpy as npimport cv2import matplotlib.pyplot as pltclass ssd(object): def __init__(self): self.num_boxes = [] self.feature_map
2020-08-22 19:44:30 393
原创 yoloV4mosaic数据增强,同步Pascal VOC格式的XML标注信息
yoloV4mosaic数据增强,同步Pascal VOC格式的XML标注信息talk is cheap, show code.talk is cheap, show code.from PIL import Image, ImageDrawimport numpy as npfrom matplotlib.colors import rgb_to_hsv, hsv_to_rgbimport mathimport xml.dom.minidomimport numpy as npimpo
2020-08-03 12:02:03 1023 8
原创 标准模板库vector
一、定义和初始化定义vector(要有头文件#include < vector >)Vector<T> v1; //默认构造函数v1为空Vector<T> v2(v1);//v2是v1的一个副本Vector<T> v3(n,i);//v3包含n个值为i的元素Vector<T> v4(n); //v4含有n个值为0的元素二、容...
2019-07-06 08:16:33 97
原创 C++ STL - queue使用详解
C++ STL - queue使用详解一:定义queue(要有头文件#include )二:基本函数三:示例代码c++队列模板类的定义在头文件中,queue 模板类需要两个模板参数,一个是元素类型,一个容器类型,元素类型是必要的,容器类型是可选的,默认为deque 类型。下面详细介绍queue的使用:一:定义queue(要有头文件#include ) queue<int> q...
2019-07-05 22:05:30 356
QT+ffmpeg+多摄像头+实时(Ubuntu直接可用)(Windows需要替换ffmpeg编解码库)
2020-10-28
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人