分类目录归档:视觉,深度学习

【置顶】ResNet18


特点 H(x) = F(x) + x

在layer1,layer2,layer3,layer4中,使用残差连接,即输入+原输出 作为新的输出, 即 out += identity

第一层:nn.Conv2d

import torch.nn as nn
# 第一个,输入通道3,输出通道64
# 如果输入图片大小为3*224*224,则输入为长宽为(224+3*2-7)/2+1=112,形状为6...

Read more

【置顶】卷积与转置卷积


1、卷积Conv2D 输出特征图的高度:

Hout=(Hin+2padding-kernel_size)/stride+1

2、转置卷积Conv2DTranspose

Hin’=Hin+(Hin−1)∗(s−1)。
k’=k
s’=1 //分母步长始终为1
p’=k-p-1
代入公式,得到
Hout=(Hin−1)∗stride−2padding+kernel_size //stride...

Read more

【置顶】深度学习中踩过的坑


安装看显卡驱动版本: nvidia-smi 然后到pytorch官网上找相应的命令

一种灵活和可移植的安装包的方法,允许指定要使用的Python解释器,即使在系统上安装了多个版本的Python。如果希望将包安装到特定的Python环境中,或者希望使用不同于系统默认Python版本,则这可能非常有用。 python.exe -s -m pip install 包名

Read more