![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
CNN
ax816
这个作者很懒,什么都没留下…
展开
-
vgg11/13/16/19-pytorch实现
vgg11/13/16/19-pytorch实现import torchimport torch.nn as nncfg = { 'A': [64, 'M', 128, 'M', 256, 256, 'M', 512, 512, 'M', 512, 512, 'M'], 'B': [64, 64, 'M', 128, 128, 'M', 256, 256, 'M', 512, 512,原创 2020-07-29 18:03:40 · 947 阅读 · 0 评论 -
python 查看网络模型
python 查看网络模型if __name__ == '__main__': import torch import torchvision.models as models from torchsummary import summary device = torch.device("cuda" if torch.cuda.is_available() else "cpu") model = models.resnet50().to(device) s原创 2020-07-29 18:01:30 · 617 阅读 · 2 评论 -
CNN基础
卷积参数量卷积参数量 = 卷积核长度x卷积核宽度x输入通道x输出通道+输出通道(偏置)卷积计算量卷积计算量 = 输出数据大小x卷积核的尺度x输入通道数例:输入:480x640x3,输出:28x28x128,卷积核:3x3卷积参数量= 3×3×3×128+128 = 3584卷积计算量 = 28×28×128×3×3×3 = 2709504...原创 2020-07-21 16:26:41 · 357 阅读 · 0 评论