tensorflow实现alexnet模型

本文介绍了如何使用TensorFlow从头构建AlexNet模型,包括模型的简化结构图,并提供了相关代码。读者可以参考此教程进行实践并进行问题交流。
摘要由CSDN通过智能技术生成

网上找了挺久没有能直接拿来用的,索性自己搭一个吧,精简结构图:
在这里插入图片描述

下面上代码:

# alexnet
# 构建模型
def inference(input_tensor, train, regularizer):
    with tf.variable_scope('layer1-conv1'):#通过tf.variable_scope函数控制tf.get_variable函数来获取以及创建过的变量
        conv1_weights = tf.get_variable("weight",[11,11,3,96],initializer=tf.truncated_normal_initializer(stddev=0.1))
        conv1_biases = tf.get_variable("bias", [96], initializer=tf.constant_initializer(0.0))
        conv1 = tf.nn.conv2d(input_tensor, conv1_weights, strides=[1, 4, 4, 1], padding='SAME')#strides=【1,1,1,1】表示滑动步长为1
        relu1 = tf.nn.relu(tf.nn.bias_add(conv1, conv1_biases))
        relu1=tf.nn.local_response_normalization(relu1, depth_radius=2, alpha=1e-4, beta=0.75, name='norm1', bias=1)

    with t
  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值