abupy.DLBu package

Submodules

abupy.DLBu.ABuDL module

abupy.DLBu.ABuDLImgStd module

深度学习工具模块,为caffe等工具库标准化图片格式

abupy.DLBu.ABuDLImgStd.std_img_from_root_dir(root_dir, a_ext)[源代码]

找到root_dir目录下所有.ext的图片,都转换其真实类型后, 将不是jpeg的全部转换为jpeg,即标准化图像格式 :param root_dir: str对象,文件夹路径信息 :param a_ext: 图像文件后缀,eg. png,jpg

abupy.DLBu.ABuDLImgStd.covert_to_jpeg(org_img, dst_img=None)[源代码]

将输入img转换为RGB的jpeg格式图像 :param org_img: 原始图像路径,str对象 :param dst_img: 转换后的图像路径输出路径,str对象,默认=None将覆盖org_img路径

abupy.DLBu.ABuDLImgStd.find_img_by_ext(ext, root_dir)[源代码]

将root_dir目录下及一级子目录下所有.ext后缀的文件全路径返回, 注意只遍历根目录及一级子目录 :param ext: str对象 图像文件后缀,eg. png,jpg :param root_dir: str对象,文件夹路径信息 :return: list序列对象

abupy.DLBu.ABuDLImgStd.change_to_real_type(img_list)[源代码]
将img的后缀名转换为其真实类型
eg. a.png 如果 a.png实际上是jpeg,则将后缀修改-> a.jpeg
参数:img_list
返回:

abupy.DLBu.ABuDLTVSplit module

深度学习工具模块,为caffe工具库做数据集准备, 切割训练集

abupy.DLBu.ABuDLTVSplit.train_val_split(train_path, n_folds=10)[源代码]

将caffe返回的数据集文件进行切割工作,切割为训练集,交叉测试集,测试集数据文件

Module contents