Skip to content

Instantly share code, notes, and snippets.

def cnn_model(features, labels, mode, params):
'''
CNN model architecture for use with TensorFlow estimator
Architecture:
256x256x3 input -> [conv -> max pool] x5 -> fully_connected -> softmax
Inputs:
features -- the input images, shape: [N x 256 x 256 x 3]
labels -- raw class integers, shape: [N, ]