資源簡介
使用tflearn高層封裝,用alexnet對鮮花數(shù)據(jù)集進(jìn)行訓(xùn)練
代碼片段和文件信息
from?__future__?import?division?print_function?absolute_import
import?tflearn
from?tflearn.layers.core?import?input_data?dropout?fully_connected
from?tflearn.layers.conv?import?conv_2d?max_pool_2d
from?tflearn.layers.normalization?import?local_response_normalization
from?tflearn.layers.estimator?import?regression
import?tflearn.datasets.oxflower17?as?oxflower17
X?Y?=?oxflower17.load_data(one_hot=True?resize_pics=(227?227))??##此句調(diào)用了tflearn文件夾下dataset中oxflower17.py函數(shù),
##此函數(shù)主要作用是下載Oxfords數(shù)據(jù)
#?Building?‘AlexNet‘
network?=?input_data(shape=[None?227?227?3])
network?=?conv_2d(network?96?11?strides=4?activation=‘relu‘)
network?=?max_pool_2d(network?3?strides=2)
network?=?local_response_normalization(network)
network?=?conv_2d(network?256?5?activation=‘rel
評論
共有 條評論