AlexNet系列 1:利用 Tensorflow 实现模型代码
发布时间
阅读量:
阅读量
目录
AlexNet
1. 训练脚本:train.py
2. 测试脚本:test.py
3. 模型定义文件:model.py
4. 图像处理模块:images.py
AlexNet
以下代码是用于 Udacity 交通信号灯检测任务的实现方案:
1. train.py
import tensorflow as tf
import model
import utils
import images
def get_loss(result, y):
cross_entropy = tf.nn.sigmoid_cross_entropy_with_logits(logits=result, labels=y)
return tf.reduce_mean(cross_entropy)
def get_optimizer(loss):
train_variables = tf.trainable_variables()
optimizer = tf.train.AdamOptimizer(learni
全部评论 (0)
还没有任何评论哟~
