![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
经典神经网络模型
文章平均质量分 67
爱吃苹果的派大星
这个作者很懒,什么都没留下…
展开
-
使用PyTorch搭建ResNet101、ResNet152网络
参照ResNet50的搭建,由于50层以上几乎相同,叠加卷积单元数即可,所以没有写注释。101和152的搭建注释可以参照我的ResNet50搭建中的注释:训练可以参照我的ResNet18搭建中的训练部分:ResNet101和152可以依旧参照ResNet50的网络图片:上代码:ResNet101的model.py模型:import torchimport torch.nn as nnfrom torch.nn import functional as Fclass DownSampl原创 2022-02-25 00:29:16 · 9546 阅读 · 1 评论 -
使用PyTorch搭建ResNet50网络
看过我之前ResNet18和ResNet34搭建的朋友可能想着可不可以把搭建18和34层的方法直接用在50层以上的ResNet的搭建中,我也尝试过。但是ResNet50以上的网络搭建不像是18到34层只要简单修改卷积单元数目就可以完成,ResNet50以上的三种网络都是一个样子,只是层数不同,所以完全可以将34到50层作为一个搭建分水岭。加上我初学PyTorch和深度神经网络,对于采用BasicBlock和BottleNeck的高效率构建还不是很懂,所以这里给出了类似前两种ResNet的简单暴力堆叠网络层原创 2022-02-25 00:26:30 · 5677 阅读 · 2 评论 -
使用PyTorch搭建ResNet34网络
ResNet34网络结构先上图参照ResNet18的搭建,由于34层和18层几乎相同,叠加卷积单元数即可,所以没有写注释,具体可以参考我的ResNet18搭建中的注释,ResNet34的训练部分也可以参照。import torchimport torch.nn as nnfrom torch.nn import functional as Fclass CommonBlock(nn.Module): def __init__(self, in_channel, out_chann原创 2022-02-25 00:23:20 · 7146 阅读 · 1 评论 -
使用PyTorch搭建ResNet18网络并使用CIFAR10数据集训练测试
ResNet18网络结构具体ResNet原理细节这里不多做描述,直接上代码这里并未采用BasicBlock和BottleNeck复现ResNet18model.py网络模型部分:import torchimport torch.nn as nnfrom torch.nn import functional as F"""把ResNet18的残差卷积单元作为一个Block,这里分为两种:一种是CommonBlock,另一种是SpecialBlock,最后由ResNet18统筹调度其中S原创 2022-02-25 00:18:29 · 8703 阅读 · 8 评论