Advertisement

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)

还没有任何评论哟~