深度学习当中的模块缝合学习笔记(一)

89 阅读1分钟
  1. 如何根据论文找到想要提取的模块
  • 先在根节点目录找model文件夹,通过model文件夹找到需要提取的模块
  • 如果找不到就找train.py文件夹找引用的部分,可以找到对应的模块位置
  • 新建一个.py文件,然后将模块代码粘贴至新文件当中,并附上如下的代码对模块代码进行调试 一般设置的是B H W C(索引为:0 1 2 3),如果给出的模块不是这个参数顺序,则需要进行调整参数测试,并且可以对模型进行更改: image.png 修改维度操作

image.png

image.png

image.png

image.png

image.png print(x.shape)