chitra 0.0.1

Library for creating data input pipeline in pure Tensorflow 2.x

Chitra

Library for creating data input pipeline in pure Tensorflow 2.x

.

Install

pip install chitra

How to use

Loading data for image classification

from chitra.dataloader import Clf

path = '/Users/aniketmaurya/Pictures/cats'

clf_dl = Clf()
data = clf_dl.from_folder(path)

print('class names:', clf_dl.CLASS_NAMES)

for e in data.take(1):
    print('image shape:', e[0].shape)
    print('class:', e[1].numpy())

class names: (b'whitecat', b'blackcat')
image shape: (1, 160, 160, 3)
class: b'whitecat'