数据处理
halo_小怪兽
视觉算法工程师
展开
-
pytorch中的nn.ZeroPad2d(),零填充函数
在卷积神经网络中,有使用设置padding的参数,配合卷积步长,可以使得卷积后的特征图尺寸大小不发生改变,那么在手动实现图片或特征图的边界零填充时,常用的函数是nn.ZeroPad2d(),可以指定tensor的四个方向上的填充,比如左边添加1dim、右边添加2dim、上边添加3dim、下边添加4dim,即指定paddin参数为(1,2,3,4),本文中代码设置的是(3,4,5,6)如下:import torch.nn as nnimport cv2import torchvisionfrom to原创 2021-04-13 10:31:48 · 9485 阅读 · 2 评论 -
计算自定义图片数据集的平均像素值python代码
在计算机视觉领域对图片进行预处理时,通常需要先计算出数据集中所有图片的平均像素均值,本文详细给出了计算自定义图片数据集的平均像素的python代码。1.获取自定义数据集的全部绝对路径def is_imgfile(filepath): filepath = os.path.expanduser(filepath) if os.path.isfile(filepath) and imghdr.what(filepath): return True return Fal原创 2021-04-11 20:57:32 · 2008 阅读 · 1 评论 -
python的基本文件操作
1. python文件操作1.1 批量对文件进行改名src_path='' #需要改名的数据集文件夹路径dst_path='' #将改名后的数据集放入的文件夹路径import os#加载数据集文件夹files=os.listdir(src_path)print("文件数量:"len(files))for i ,filename in enumerate(files): #print(filename) src=os.path.join(os.path.abspath(s原创 2020-12-01 20:14:37 · 220 阅读 · 0 评论