NI Deep - Deep Learning Tools from the NI Group at TU-Berlin

Build Status Coverage Status License Code Climate Codacy Badge

This is a collection of scripts, utilities and examples to use with caffe. These are either too use case specific and are therefore kept outside of the Caffe framework. They may also be considered completely independent and potentially useful in combination with other frameworks.

Use cases include:

Disclaimer:

Some automated testing is in place. However, do not assume the functionality is free of bugs. Feel free to inspect them, scrutinize them. Feedback on making them more usable for others is welcome.

Beware: The documentation is pretty scarce. We're working on it.

Getting Started:

Dependencies:

Examples:

PASCAL-Context to LMDB

Download val_59.txt from here.

Download 59_labels.txt from here.

from nideep.datasets.pascal_context_to_lmdb import pascal_context_to_lmdb
val_list_path = os.path.expanduser('~/data/PASCAL-Context/val_59.txt')
with open(val_list_path, 'r') as f:
    val_list = f.readlines()
    val_list = [l.translate(None, ''.join('\n')) for l in val_list if len(l) > 0]

nt, nv, fpath_imgs_train, fpath_labels_train, fpath_imgs_val, fpath_labels_val = \
pascal_context_to_lmdb(os.path.expanduser('~/data/VOCdevkit/VOC2012/JPEGImages'),
                       os.path.expanduser('~/data/PASCAL-Context/trainval'),
                       os.path.expanduser('~/data/PASCAL-Context/labels.txt'),
                       os.path.expanduser('~/data/PASCAL-Context/59_labels.txt'),
                       '',
                       os.path.expanduser('~/data/PASCAL-Context/'),
                       val_list=val_list
                       )

print "size: %d" % nt, nv, fpath_imgs_train, fpath_labels_train, fpath_imgs_val, fpath_labels_val