コード例 #1
0
ファイル: generatelmdb.py プロジェクト: pierrepayen/PFECaffe
def generate_lmdb(lmdb_path, datum_id_list, datums_file,
                  labels_file, min_value, max_value):
    lmdb_manager = LMDBManager(lmdb_path)
    n_channels = 1
    for datum_line, label_line in zip(datums_file, labels_file):
        x = np.fromstring(datum_line)
        y = int(label_line)
        x = x.reshape(n_channels, n_vectors, n_components)
        lmdb_manager.add_data(datum_id, x, y, min_value, max_value)
コード例 #2
0
ファイル: readlmdb.py プロジェクト: pierrepayen/PFECaffe
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Provides an example of reading informations out of a LMDB database.
"""


import sys
from lmdbmanager import LMDBManager

if __name__ == "__main__":
    lmdb_path = sys.argv[1]
    # lmdb_path = "mylmdb"
    # lmdb_path = "/data1/home/adam.oudad/Caffe/caffe-master/examples/mnist/mnist_test_lmdb"
    # lmdb_path = "/data1/home/adam.oudad/Caffe/tuned-networks/lenet-kaldi/kaldi_lmdb_fault"
    lmdb_manager = LMDBManager(lmdb_path)

    print lmdb_manager.info()
    datum = lmdb_manager.get_by_id(56)
    print lmdb_manager.convert_datum(datum)
    lmdb_manager.print_datum_info(56)
    lmdb_manager.render_datum(datum)
    print lmdb_manager.get_labels()
    lmdb_manager.close()