OCR-project/dataset_8h_source.html

#ifndef DATASET_H

#define DATASET_H


#include "cnn.h"

#include <stddef.h>


typedef struct {

    float pixels[CNN_IMG_H * CNN_IMG_W];

    int   label;

} Sample;


typedef struct {

    Sample *data;

    size_t  size;

    size_t  cap;

} Dataset;


Dataset *dataset_load(const char *root_dir, int n_threads);


void dataset_shuffle(Dataset *ds);


void dataset_free(Dataset *ds);


void dataset_print_info(const Dataset *ds);


#endif /* DATASET_H */

cnn.h
Convolutional Neural Network for A–Z character recognition.

CNN_IMG_W
#define CNN_IMG_W
Definition cnn.h:30

CNN_IMG_H
#define CNN_IMG_H
Definition cnn.h:29

dataset_load
Dataset * dataset_load(const char *root_dir, int n_threads)
Load all images from root_dir into a Dataset.
Definition dataset.c:128

dataset_shuffle
void dataset_shuffle(Dataset *ds)
Shuffle the samples in a Dataset in-place (Fisher-Yates).
Definition dataset.c:207

dataset_free
void dataset_free(Dataset *ds)
Free all memory owned by a Dataset.
Definition dataset.c:221

dataset_print_info
void dataset_print_info(const Dataset *ds)
Print a summary of the dataset to stdout.
Definition dataset.c:229

Dataset
A collection of training samples.
Definition dataset.h:39

Dataset::data
Sample * data
Definition dataset.h:40

Dataset::size
size_t size
Definition dataset.h:41

Dataset::cap
size_t cap
Definition dataset.h:42

Sample
A single labelled training sample.
Definition dataset.h:29

Sample::label
int label
Definition dataset.h:33

Sample::pixels
float pixels[CNN_IMG_H *CNN_IMG_W]
Definition dataset.h:31