手写四维卷积,python and C

2023-11-11 13:18
文章标签 python 卷积 手写 四维

本文主要是介绍手写四维卷积,python and C,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

手写四维卷积,python and C

  • 数据结构
    • C:
    • Python:

数据结构

数据:NHWC
权重:OIHW

C:

void convOp(Blob input,Blob* output,conv_s convInfo, wtParam param){int sh = convInfo.strideH, sw = convInfo.strideW,ph = convInfo.convPad.padH, pw = convInfo.convPad.padW,kh = convInfo.kernelH, kw = convInfo.kernelW;Blob padedblob;padedblob.n = input.n;padedblob.h = input.h + 2 * ph;padedblob.w = input.w + 2 * pw;padedblob.c = input.c;padedblob.dataFmt = input.dataFmt;padedblob.dataType = input.dataType;padedblob.data = malloc(sizeof(float) * padedblob.n * padedblob.h * padedblob.w * padedblob.c);memset(padedblob.data, 0, sizeof(float) * padedblob.n * padedblob.h * padedblob.w * padedblob.c);//补边for(int n = 0; n < input.n; n++){for(int c = 0; c < input.c; c++){for(int h = 0; h < input.h; h++){for(int w = 0; w < input.w; w++){blobSet(n,h + ph,w + pw,c,padedblob,blobGet(n,h,w,c,input));}}}}output->data = malloc(sizeof(float) * output->n * output->h * output->w * output->c);memset(output->data, 0, sizeof(float) * output->n * output->h * output->w * output->c);//卷积计算for(int n = 0; n < padedblob.n; n++){for(int oc = 0; oc < param.n; oc++){// wt[oc, 256, 3, 3]for(int inh = 0; inh <= padedblob.h - kh; inh += sh){for(int inw = 0; inw <= padedblob.w - kw; inw += sw){float sum = 0;for(int inc = 0; inc < padedblob.c; inc++){for(int offx = 0; offx < kh; offx++){for(int offy = 0; offy < kw; offy++){sum += blobGet(n, inh + offx, inw + offy, inc, padedblob) *wtGet(oc, offx, offy, inc, param);}}}blobSet(n, inh / sh, inw / sw, oc, *output, sum);}}}}}

Python:

def convOp(inblob, node_size, node_shape, param):"""param : OIHWreturn: outblob->nparray"""param = param[0]kh = int(node_shape[1][0])kw = int(node_shape[1][1])sh = int(node_shape[2][0])sw = int(node_shape[2][1])ph = int(node_shape[3][0])pw = int(node_shape[3][1])if ph != 0 or pw != 0:# 补边儿newn = inblob.shape[0]newh = inblob.shape[1] + 2 * phneww = inblob.shape[2] + 2 * pwnewc = inblob.shape[3]paddedblob = np.zeros((newn, newh, neww, newc), dtype=np.float32)paddedblob[:, ph:ph + inblob.shape[1], pw:pw + inblob.shape[2], :] = inblobelse:paddedblob = inbloboutblob = np.zeros(tuple(node_size), dtype=np.float32)# 权重是OIHW,而数据是NHWC,为了H与W能够在numpy中通过广播的方式计算,需要把权重转置成OHWI,numpy的转置几乎不消耗时间,因为该转置是不改变数据在内存中表示的。# 权重:OIHW 		OHWI# 数据:   	     	NHWCparam = np.transpose(param, [0, 2, 3, 1])#卷积计算for oc in range(param.shape[0]):  # O of weightfor h in range(outblob.shape[1]):  # H of outblobfor w in range(outblob.shape[2]):outblob[:, h, w, oc] = np.sum(paddedblob[:,h * sh: h * sh + param.shape[1],w * sw: w * sw + param.shape[2],:] * param[oc, :, :, :])return outblob

这篇关于手写四维卷积,python and C的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/390318

相关文章

基于Python实现一个图片拆分工具

《基于Python实现一个图片拆分工具》这篇文章主要为大家详细介绍了如何基于Python实现一个图片拆分工具,可以根据需要的行数和列数进行拆分,感兴趣的小伙伴可以跟随小编一起学习一下... 简单介绍先自己选择输入的图片,默认是输出到项目文件夹中,可以自己选择其他的文件夹,选择需要拆分的行数和列数,可以通过

Python中反转字符串的常见方法小结

《Python中反转字符串的常见方法小结》在Python中,字符串对象没有内置的反转方法,然而,在实际开发中,我们经常会遇到需要反转字符串的场景,比如处理回文字符串、文本加密等,因此,掌握如何在Pyt... 目录python中反转字符串的方法技术背景实现步骤1. 使用切片2. 使用 reversed() 函

Python中将嵌套列表扁平化的多种实现方法

《Python中将嵌套列表扁平化的多种实现方法》在Python编程中,我们常常会遇到需要将嵌套列表(即列表中包含列表)转换为一个一维的扁平列表的需求,本文将给大家介绍了多种实现这一目标的方法,需要的朋... 目录python中将嵌套列表扁平化的方法技术背景实现步骤1. 使用嵌套列表推导式2. 使用itert

使用Docker构建Python Flask程序的详细教程

《使用Docker构建PythonFlask程序的详细教程》在当今的软件开发领域,容器化技术正变得越来越流行,而Docker无疑是其中的佼佼者,本文我们就来聊聊如何使用Docker构建一个简单的Py... 目录引言一、准备工作二、创建 Flask 应用程序三、创建 dockerfile四、构建 Docker

Python使用vllm处理多模态数据的预处理技巧

《Python使用vllm处理多模态数据的预处理技巧》本文深入探讨了在Python环境下使用vLLM处理多模态数据的预处理技巧,我们将从基础概念出发,详细讲解文本、图像、音频等多模态数据的预处理方法,... 目录1. 背景介绍1.1 目的和范围1.2 预期读者1.3 文档结构概述1.4 术语表1.4.1 核

Python使用pip工具实现包自动更新的多种方法

《Python使用pip工具实现包自动更新的多种方法》本文深入探讨了使用Python的pip工具实现包自动更新的各种方法和技术,我们将从基础概念开始,逐步介绍手动更新方法、自动化脚本编写、结合CI/C... 目录1. 背景介绍1.1 目的和范围1.2 预期读者1.3 文档结构概述1.4 术语表1.4.1 核

Conda与Python venv虚拟环境的区别与使用方法详解

《Conda与Pythonvenv虚拟环境的区别与使用方法详解》随着Python社区的成长,虚拟环境的概念和技术也在不断发展,:本文主要介绍Conda与Pythonvenv虚拟环境的区别与使用... 目录前言一、Conda 与 python venv 的核心区别1. Conda 的特点2. Python v

Python使用python-can实现合并BLF文件

《Python使用python-can实现合并BLF文件》python-can库是Python生态中专注于CAN总线通信与数据处理的强大工具,本文将使用python-can为BLF文件合并提供高效灵活... 目录一、python-can 库:CAN 数据处理的利器二、BLF 文件合并核心代码解析1. 基础合

Python使用OpenCV实现获取视频时长的小工具

《Python使用OpenCV实现获取视频时长的小工具》在处理视频数据时,获取视频的时长是一项常见且基础的需求,本文将详细介绍如何使用Python和OpenCV获取视频时长,并对每一行代码进行深入解析... 目录一、代码实现二、代码解析1. 导入 OpenCV 库2. 定义获取视频时长的函数3. 打开视频文

Python中你不知道的gzip高级用法分享

《Python中你不知道的gzip高级用法分享》在当今大数据时代,数据存储和传输成本已成为每个开发者必须考虑的问题,Python内置的gzip模块提供了一种简单高效的解决方案,下面小编就来和大家详细讲... 目录前言:为什么数据压缩如此重要1. gzip 模块基础介绍2. 基本压缩与解压缩操作2.1 压缩文