F.conv2d实现代码

import math
import torch
import torch.nn.functional as Fdef conv(x, kernel, bais, stride, padding):'''x: n,cin,h,wkernel: cout, cin, kh, kw'''n,c,h,w = x.size()          #获得x原始尺度信息x = F.pad(x, (padding,)*4) #获得kernel参数，计算输出特征图大小和x上划区域的大小cout, cin, kh, kw = kernel.size()oh = math.floor((h - kh + 2*padding)/stride + 1)  #输出特征图大小，用x原始尺度计算ow =  math.floor((w - kw + 2*padding)/stride + 1) #输出特征图大小，用x原始尺度计算out = torch.zeros(n,cout,oh, ow)  #定义输出特征图for i in range(0, h-kh+1, stride):for j in range(0, w-kw+1, stride):region = x[:,:,i:i+kh, j:j+kw] #x上取特征大小为n,cin,kw,kwfor k in range(cout):  #遍历输出个数个卷积核out[:,k,int(i/stride),int(j/stride)] = torch.sum(region * kernel[k]) # n,cin,kh, kw * None, cin, kh,kwprint(out)
if __name__ == '__main__':x = torch.rand(1,3,5,5)kernel = torch.rand(4,3,3,3)conv(x, kernel, None, 2, 1)r = F.conv2d(x, kernel, bias=None, stride=2, padding=1)print(r)

F.conv2d实现代码相关推荐

一次将tensorflow中的tf.nn.conv2d优(re)雅(shape)的转化成torch.nn.F.conv2d的咸鱼操作
写在前面毕设勉强中-看了<A Capsule Network-based Embedding Model for Knowledge Graph Completion and Search P ...
matlab 1 f噪声,跪求1/f噪声生成代码解释！！
以下是我查到的1/fnoise生成matlab代码,但是看不懂dbc_per_hz和num_taps是怎么设定的,求高手解答!!谢谢!! % function [pn, theta] = phase_ ...
vscode怎么对选定的代码格式化？ctrl+k,ctrl+f（格式化代码）
先选中代码: 然后按CTRL + K 再按CTRL+F 也可以先选择要格式化的代码块,ctrl+shift+p,搜索format,然后第二个就是:
2023美赛F题全部代码+数据+结果数学建模
2023年美赛F题全部思路数据代码都已完成全部内容见链接:https://www.jdmm.cc/file/2708700/ 1.根据文献选的GGDP的指标,发现GGDP与水资源等有关,由此可以筛 ...
VScode格式化代码快捷键shift+alt+F，格式化代码插件推荐，修改快捷键
效果图推荐插件 beautify 具体步骤打开"扩展"板块搜索安装插件"beautify" 打开需要格式化的文档使用快捷键"shift+alt ...
2020研究生数学建模大赛F题思路代码实现
代码已提交仅供交流参考赛题 2020年A题–华为题_芯片相噪算法设计.docx 2020年B题–汽油辛烷值建模.zip 2020年C题–面向康复工程的脑电信号分析和判别模型.docx 附件1-P3 ...
基于torch.nn.functional.conv2d实现CNN
在我们之前的实验中,我们一直用torch.nn.Conv2D来实现卷积神经网络,但是torch.nn.Conv2D在实现中是以torch.nn.functional.conv2d为基础的,这两者的区别 ...
输出分组_通过分组卷积的思想，巧妙的代码实现动态卷积(Dynamic Convolution)
论文的题目为<Dynamic Convolution: Attention over Convolution Kernels> paper的地址https://arxiv.org/pdf/ ...
论文翻译：搜索人脸活体检测的中心差异卷积网络及实现代码
搜索人脸活体检测的中心差异卷积网络摘要 1. 绪论 2. 相关工作人脸活体检测卷积运算符神经架构搜索 3. 方法论 3.1 中心差分卷积基本卷积基本卷积结合中心差分操作中心差分卷积的实现 ...
基于实例分割方法的端到端车道线检测论文+代码解读
Towards End-to-End Lane Detection: an Instance Segmentation Approach 论文原文 https://arxiv.org/pdf/1802 ...

F.conv2d实现代码

F.conv2d实现代码相关推荐

最新文章

热门文章