![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
好用的python小程序
Fly~~
这个作者很懒,什么都没留下…
展开
-
YOLOV4用训练好的模型实现在测试集上批量画框
很多现存的YOLOV4开源代码是使用单张图片进行测试,下面这个程序可以根据训练好的模型实现对测试集进行批量画框,简单方便。import osimport numpy as npimport copyimport colorsysimport timefrom timeit import default_timer as timerfrom keras import backend as Kfrom keras.models import load_modelfrom keras.layer原创 2022-03-18 15:28:36 · 553 阅读 · 1 评论 -
python实现K-means锚框聚类
根据标签文件XML中目标框的坐标,聚类出合适数据集的锚框,anchors_num = 9,设置聚类出的锚框个数是9个,更改路径即可直接使用。import numpy as npimport xml.etree.ElementTree as ETimport globimport randomdef cas_iou(box,cluster): x = np.minimum(cluster[:,0],box[0]) y = np.minimum(cluster[:,1],box[1原创 2022-03-18 15:13:44 · 887 阅读 · 1 评论 -
python根据x轴、y轴坐标在坐标轴里画出曲线图
python根据x轴、y轴坐标在坐标轴里画出曲线图程序实现:import numpy as npimport matplotlib as mplimport matplotlib.pyplot as plta=[1,2,3,5,6,10]b=[0,3,4,5,7,8]c=[10,9,8,7,6,5,2,0]d=[1,2,3,5,5,6,4,0]plt.title('Title')plt.plot(a, b, color='blue', label='test1')plt.plot原创 2022-03-01 10:55:12 · 7058 阅读 · 0 评论 -
批量重命名文件夹中的图片文件
只需要修改相应路径即可直接使用,简单方便!!!!# -*- coding:utf8 -*-import osclass name_jpg(): ''' 批量重命名文件夹中的图片文件 ''' def __init__(self): self.path = r'D:/algorithm/train_resize/xml' #表示需要命名处理的文件夹 def rename(self): filelist = os.listdir.原创 2021-03-05 11:38:50 · 318 阅读 · 0 评论 -
随机划分训练集、测试集
只需要把root_path 的路径修改即可# -*- coding: utf-8 -*-from __future__ import division, print_function, absolute_importimport syssys.path.append('../../')import shutilimport osimport randomimport mathdef mkdir(path): if not os.path.exists(path): .原创 2021-01-29 18:21:33 · 270 阅读 · 0 评论 -
python实现根据文件名删除文件
根据txt文档里的文件名,删除相关的内容,txt文档里可以是图片的名,也可以是xml名,如下示例;# -*- coding: utf-8 -*-import osdef delete_file(dirname): filelist = os.listdir(dirname) for file in filelist: oldpath = os.path.join(dirname, file) filename = os.path.splitext(.原创 2021-01-29 18:17:47 · 2275 阅读 · 0 评论 -
根据xml文件中目标坐标计算目标占比、长宽、面积等
下面程序主要用来统计:计算每一个目标在原图中的占比;计算目标的平均长度;计算平均宽度;计算平均面积;计算目标平均占比;# -*- coding:utf-8 -*-import osimport xml.etree.ElementTree as ETimport numpy as np#np.set_printoptions(suppress=True, threshold=np.nan) #10,000,000np.set_printoptions(suppress=True, .原创 2021-01-25 15:07:52 · 917 阅读 · 0 评论 -
将xml中目标框绘制到原图上
做目标检测的首要前提是对自己的数据了如指掌,有时候使用其他人的数据集,第一时间可能不知道其他人的打标签标准是什么,那么我们可以可视出来目标框,把标签文件里的坐标绘制到原图上,下面程序只需要给定图片、标签和保存位置,就可以直接使用。# -*- coding: utf-8 -*-from __future__ import divisionimport osimport xml.dom.minidomimport cv2import sysimport numpy as np# from im.原创 2021-01-25 14:59:25 · 1067 阅读 · 0 评论 -
目标检测中计算数据集每一类别框的数量
众所周知,一个好的目标检测模型需要大量的数据来训练,当数据量较多的情况,我们没办法直观看到每一类别的目标框个数,就无法判断类别是否平衡。下面的算法作用就是统计数据集中每一类别的目标框个数,只需要修改xml文件地址就可直接使用。# -*- coding:utf-8 -*-import osimport xml.etree.ElementTree as ETimport numpy as npimport matplotlibfrom PIL import Imagedef parse_ob.原创 2021-01-25 14:40:36 · 2910 阅读 · 0 评论