tensorlfow实现ZFNET
实现ZFNET网络的训练与验证部分 ZFNET 对ALexNet做了一些改进,但并不大,意义是可视化了卷积神经网络的内部参数。 本文训练使用的数据集是cifar-10。 网络架构(zfnet.py) import tensorflow as tf class ZFNet : def __init__ ( self , input_width = 224 , input_height = 224 , input_channels = 3 , num_classes = 1000 , learning_rate = 0.01 , momentum = 0.9 , keep_prob = 0.5 ) : self . input_width = input_width self . input_height = input_height self . input_channels = input_channels self . num_classes = num_classes self . learning_rate = learning_rate self . momentum = momentum self . keep_prob = keep_prob self . random_mean = 0 self . random_stddev = 0.01 with tf .