62、python - 全手写搭建 resnet50 神经网络

2024-01-07 21:28:50

如果将上篇文章中涉及到的算法都自己手写完一遍后,我们开始尝试利用自己手写的算法,搭建一个完整的 resnet50 神经网络。

网络结构就参考这个链接中的网络结构,是在之前下载模型的章节中,下载的模型。

为了搭建一个完整的神经网络,定义一个 Resnet 的类,这个类就包含 resnet50 的所有层和结构。该类很简单,只有一个名为run的函数,用来运行这个神经网络。

下面是 python 代码定义的这个类。

class Resnet():
  def run(self, img):
    out = ComputeConvLayer(img, "conv1")
    out = ComputeBatchNormLayer(out, "bn1")
    out = ComputeReluLayer(out)
    out = ComputeMaxPoolLayer(out)

    # layer1 
    out = ComputeBottleNeck(out, "layer1_bottleneck0", down_sample = True)
    out = Co

文章来源:https://blog.csdn.net/dongtuoc/article/details/135444353
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。