visionner

Visionner turn raw image data into numpy array, more suitable for deep learning task

Downloads
55
Stars
10
Committers
2

Installation

  • step 1
git clone https://github.com/charleslf2/visionner
  • step 2
cd visionner
  • step 3
py setup.py install

Usage


>>> from visionner.core import DatasetImporter

>>> your_dataset=DatasetImporter("path/to/your/dataset/", size=(28, 28))


>>> from visionner.core import SupervisedImporter

>>> features, labels= SupervisedImporter("path/to/your/dataset", categories=["cat", "dog"], size=(28,28))


### normalize your dataset

>>> from visionner.core import DatasetNormalizer

>>> your_normalized_dataset=DatasetNormalizer(your_dataset)


### create a trainset and a testset

>>> from visionner.core import TrainTestSpliter

>>> x_train, x_test=TrainTestSpliter(dataset, test_size=0.2)


### visualize the first image of your dataset

>>> import matplotlib.pyplot as plt 

>>> plt.imshow(your_dataset[0])
>>> plt.show()


### save your dataset

>>> from visionner.core import DatasetSaver

>>> DatasetSaver("my_saved_dataset", your_dataset)


### open your dataset

>>> from visionner.core import DatasetOpener

>>> my_saved_dataset=DatasetOpener("my_saved_dataset.npy")