def get_model(width, height, classes=40): # TODO, modify model # Building 'VGG Network' network = input_data(shape=[None, width, height, 1]) # if RGB, 224,224,3 network = conv_2d(network, 64, 3, activation='relu') #network = conv_2d(network, 64, 3, a…