图像分割实践之模型篇

内容分享2小时前发布
0 0 0

视频讲解移步 西瓜视频模型篇

模型构建三步走

  • 挖锄簸箕拿着走;
  • 先拿节点再组建;
  • 最后输出看运气;

第一步:

拿出骨干等工具

图像分割实践之模型篇

第二步:

节点我要feat二三四

图像分割实践之模型篇

第三步:

输出我选list

图像分割实践之模型篇

模型就愉快地出来了

图像分割实践之模型篇

基础知识

Pytorch搭建模型常用的3种方式如下:

  • 1.使用 nn.Sequential 直接按顺序构建的模型;
  • 2.继承 nn.Module 基类构建自定义模型;
  • 3.继承 nn.Module 基类,通过辅助模型容器(Sequential, ModuleList, ModuleDict)来构建。

辅助模型容器如下:

  • nn.Sequential:顺序性,各网络层之间严格按顺序执行,用于固定block构建;
  • nn.ModuleList:迭代性,list形式,用于大量重复网络层构建,通过for循环实现重复构建;
  • nn.ModuleDict:索引性,dict形式,用于可选择的网络层,key+value调用方式;

部分基础模块(其他可参考 >> Pytorch.nn):

  • torch.nn.Conv2d
  • torch.nn.MaxPool2d, torch.nn.AvgPool2d
  • torch.nn.BatchNorm2d
  • torch.nn.ReLU

视频讲解

模型篇

代码讲解

模型篇

© 版权声明

相关文章

暂无评论

none
暂无评论...