Beispiel #1
0
def create_dataset():
    """
    Create the dataset file.
    """
    path = "/data/lisatmp/data/faces_bbox/"
    out_file = "test_outputmap.h5"
    outputmap_file = OutputMapFile(path=path)
    n_examples=3000
    out_shape=(3000, 100, 100)
    h5file, gcols = OutputMapFile.create_file(path=path, filename=out_file, n_examples=n_examples, out_shape=out_shape)

    pt = numpy.random.randint(255, size=out_shape)
    ino = numpy.random.rand(n_examples)
    iloc = numpy.random.rand(n_examples)
    tgt = numpy.random.rand(n_examples)

    outputmap_file.save_output_map(h5file, pt , ino, iloc, tgt)
    h5file.close()