CIFAR10CNNFlask GitHub Details, Stars and Alternatives

Loading repository data…

README preview

Building a convolutional neural network (CNN/ConvNet) using TensorFlow NN (tf.nn) module. The CNN model architecture is created and trained using the CIFAR10 dataset. The model is accessed using HTTP by creating a Web application using Python and Flask. The following diagram summarizes the project. system_diagram

The project steps are as follows:

The training data should be read and prepared for use with the CNN.

def get_dataset_images(dataset_path, im_dim=32, num_channels=3):
    """
    This function accepts the dataset path, reads the data, and returns it after being reshaped to match the requierments of the CNN.
    :param dataset_path:Path of the CIFAR10 dataset binary files.
    :param im_dim:Number of rows and columns in each image. The image is expected to be rectangular.
    :param num_channels:Number of color channels in the image.
    :return:Returns the input data after being reshaped and output labels.
    """
    num_files = 5#Number of training binary files in the CIFAR10 dataset.
    images_per_file = 10000#Number of samples withing each binary file.
    files_names = os.listdir(patches_dir)#Listing the binary files in the dataset path.
    """
    Creating an empty array to hold the entire training data after being reshaped.
    The dataset has 5 binary files holding the data. Each binary file has 10,000 samples. Total number of samples in the dataset is 5*10,000=50,000.
    Each sample has a total of 3,072 pixels. These pixels are reshaped to form a RGB image of shape 32x32x3.
    Finally, the entire dataset has 50,000 samples and each sample of shape 32x32x3 (50,000x32x32x3).
    """
    dataset_array = numpy.zeros(shape=(num_files * images_per_file, im_dim, im_dim, num_channels))
    #Creating an empty array to hold the labels of each input sample. Its size is 50,000 to hold the label of each sample in the dataset.
    dataset_labels = numpy.zeros(shape=(num_files * images_per_file), dtype=numpy.uint8)
    index = 0#Index variable to count number of training binary files being processed.
    for file_name in files_names:
        """
        Because the CIFAR10 directory does not only contain the desired training files and has some  other files, it is required to filter the required files.
        Training files start by 'data_batch_' which is used to test whether the file is for training or not.
        """
        if file_name[0:len(file_name) - 1] == "data_batch_":
            print("Working on : ", file_name)
            """
            Appending the path of the binary files to the name of the current file.
            Then the complete path of the binary file is used to decoded the file and return the actual pixels values.
            """
            data_dict = unpickle_patch(dataset_path+file_name)
            """
            Returning the data using its key 'data' in the dictionary.
            Character b is used before the key to tell it is binary string.
            """
            images_data = data_dict[b"data"]
            #Reshaping all samples in the current binary file to be of 32x32x3 shape.
            images_data_reshaped = numpy.reshape(images_data, newshape=(len(images_data), im_dim, im_dim, num_channels))
            #Appending the data of the current file after being reshaped.
            dataset_array[index * images_per_file:(index + 1) * images_per_file, :, :, :] = images_data_reshaped
            #Appening the labels of the current file.
            dataset_labels[index * images_per_file:(index + 1) * images_per_file] = data_dict[b"labels"]
            index = index + 1#Incrementing the counter of the processed training files by 1 to accept new file.
    return dataset_array, dataset_labels#Returning the training input data and output labels.

def unpickle_patch(file):
    """
    Decoding the binary file.
    :param file:File to decode it data.
    :return: Dictionary of the file holding details including input data and output labels.
    """
    patch_bin_file = open(file, 'rb')#Reading the binary file.
    patch_dict = pickle.load(patch_bin_file, encoding='bytes')#Loading the details of the binary file into a dictionary.
    return patch_dict#Returning the dictionary.

def create_conv_layer(input_data, filter_size, num_filters): """ Builds the CNN convolution (conv) layer. :param input_data:patch data to be processed. :param filter_size:#Number of rows and columns of each filter. It is expected to have a rectangular filter. :param num_filters:Number of filters. :return:The last fully connected layer of the network. """ """ Preparing the filters of the conv layer by specifiying its shape. Number of channels in both input image and each filter must match. Because number of channels is specified in the shape of the input image as the last value, index of -1 works fine. """ filters = tensorflow.Variable(tensorflow.truncated_normal(shape=(filter_size, filter_size, tensorflow.cast(input_data.shape[-1], dtype=tensorflow.int32), num_filters), stddev=0.05)) print("Size of conv filters bank : ", filters.shape) """ Building the convolution layer by specifying the input data, filters, strides along each of the 4 dimensions, and the padding. Padding value of 'VALID' means the some borders of the input image will be lost in the result based on the filter size. """ conv_layer = tensorflow.nn.conv2d(input=input_data, filter=filters, strides=[1, 1, 1, 1], padding="VALID") print("Size of conv result : ", conv_layer.shape) return filters, conv_layer#Returing the filters and the convolution layer result. def create_CNN(input_data, num_classes, keep_prop): """ Builds the CNN architecture by stacking conv, relu, pool, dropout, and fully connected layers. :param input_data:patch data to be processed. :param num_classes:Number of classes in the dataset. It helps determining the number of outputs in the last fully connected layer. :param keep_prop:probability of dropping neurons in the dropout layer. :return: last fully connected layer. """ #Preparing the first convolution layer. filters1, conv_layer1 = create_conv_layer(input_data=input_data, filter_size=5, num_filters=4) """ Applying ReLU activation function over the conv layer output. It returns a new array of the same shape as the input array. """ relu_layer1 = tensorflow.nn.relu(conv_layer1) print("Size of relu1 result : ", relu_layer1.shape) """ Max pooling is applied to the ReLU layer result to achieve translation invariance. It returns a new array of a different shape from the the input array relative to the strides and kernel size used. """ max_pooling_layer1 = tensorflow.nn.max_pool(value=relu_layer1, ksize=[1, 2, 2, 1], strides=[1, 1, 1, 1], padding="VALID") print("Size of maxpool1 result : ", max_pooling_layer1.shape) #Similar to the previous conv-relu-pool layers, new layers are just stacked to complete the CNN architecture. #Conv layer with 3 filters and each filter is of sisze of 5x5. filters2, conv_layer2 = create_conv_layer(input_data=max_pooling_layer1, filter_size=7, num_filters=3) relu_layer2 = tensorflow.nn.relu(conv_layer2) print("Size of relu2 result : ", relu_layer2.shape) max_pooling_layer2 = tensorflow.nn.max_pool(value=relu_layer2, ksize=[1, 2, 2, 1], strides=[1, 1, 1, 1], padding="VALID") print("Size of maxpool2 result : ", max_pooling_layer2.shape) #Conv layer with 2 filters and a filter sisze of 5x5. filters3, conv_layer3 = create_conv_layer(input_data=max_pooling_layer2, filter_size=5, num_filters=2) relu_layer3 = tensorflow.nn.relu(conv_layer3) print("Size of relu3 result : ", relu_layer3.shape) max_pooling_layer3 = tensorflow.nn.max_pool(value=relu_layer3, ksize=[1, 2, 2, 1], strides=[1, 1, 1, 1], padding="VALID") print("Size of maxpool3 result : ", max_pooling_layer3.shape) #Adding dropout layer before the fully connected layers to avoid overfitting. flattened_layer = dropout_flatten_layer(previous_layer=max_pooling_layer3, keep_prop=keep_prop) #First fully connected (FC) layer. It accepts the result of the dropout layer after being flattened (1D). fc_resultl = fc_layer(flattened_layer=flattened_layer, num_inputs=flattened_layer.get_shape()[1:].num_elements(), num_outputs=200) #Second fully connected layer accepting the output of the previous fully connected layer. Number of outputs is equal to the number of dataset classes. fc_result2 = fc_layer(flattened_layer=fc_resultl, num_inputs=fc_resultl.get_shape()[1:].num_elements(), num_outputs=num_classes) print("Fully connected layer results : ", fc_result2) return fc_result2#Returning the result of the last FC layer. def dropout_flatten_layer(previous_layer, keep_prop): """ Applying the dropout layer. :param previous_layer: Result of the previous layer to the dropout layer. :param keep_prop: Probability of keeping neurons. :return: flattened array. """ dropout = tensorflow.nn.dropout(x=previous_layer, keep_prob=keep_prop) num_features = dropout.get_shape()[1:].num_elements() layer = tensorflow.reshape(previous_layer, shape=(-1, num_features))#Flattening the results. return layer def fc_layer(flattened_layer, num_inputs, num_outputs): """ uilds a fully connected (FC) layer. :param flattened_layer: Previous layer after being flattened. :param num_inputs: Number of inputs in the previous layer. :param num_outputs: Number of outputs to be returned in such FC layer. :return: """ #Preparing the set of weights for the FC layer. It depends on the number of inputs and number of outputs. fc_weights = tensorflow.Variable(tensorflow.truncated_normal(shape=(num_inputs, num_outputs), stddev=0.05)) #Matrix multiplication between the flattened array and the set of weights. fc_resultl = tensorflow.matmul(flattened_layer, fc_weights) return fc_resultl#Output of the FC layer (result of matrix multiplication).

CIFAR10CNNFlask

Live repository statistics

Limited signals

README preview