Skip to content

EasyNN is a python package designed to provide an easy-to-use Neural Network. The package is designed to work right out of the box, while also allowing the user to customize features as they see fit.

License

Notifications You must be signed in to change notification settings

danielwilczak101/EasyNN

Repository files navigation

EasyNN - Neural Networks made Easy

EasyNN is a python package designed to provide an easy-to-use Neural Network. The package is designed to work right out of the box with multiple datasets, while also allowing the user to customize features as they see fit.

EasyNN requires Python version 3.9.7 or greater.

See our wiki for more information and Datasets.

Installation:

Run python's pip3 to install:

pip3 install EasyNN

Model:

from EasyNN.examples.mnist.number.trained import model

# Classify an image.
print(model.classify(image))

Dataset Example:

from EasyNN.examples.mnist.number.trained import model
from EasyNN.examples.mnist.number.data import dataset

images, labels = dataset

# Classify what the second image is in the dataset.
print(model.classify(images[1]))

# Show the image.
model.show(images[1])

Dataset example output:

Downloading - number_parameters.npz:
[################################] 1769/1769 - 00:00:00
Downloading - number_structure.pkl:
[################################] 10700/10700 - 00:00:00
Downloading - number_dataset.npz:
[################################] 11221/11221 - 00:00:00
0

Full example:

More info can be found about converting images in the utilities section.

from EasyNN.examples.mnist.number.trained import model
from EasyNN.utilities import Preprocess, download

# Download an example image.
download("three.jpg","https://bit.ly/3dbO1eV")

format_options = dict(
    grayscale=True,
    invert=True,
    process=True,
    contrast=30,
    resize=(28, 28),
    rotate=3,
)

# Converting your image into the correct format for the mnist number dataset.
image = Preprocess("three.jpg").format(**format_options)

# Classify what the image is using the pretrained model.
print(model.classify(image))

# Show the image after it has been processed.
model.show(image)

Output:

Downloading - four.jpg:
[################################] 1371/1371 - 00:00:00
3

Image output:

Trained Models

Use the trained models section to see EasyNN's datasets and pre-trained neural networks ready to run.

MNIST Number Classifier network for images of handwritten single digits between 0 and 9.
MNIST Fashion Classifier network for ten classes of human clothing images of the size 28x28 pixels.
Cifar 10 Classifier network for ten types of images varying from airplane, cat, dog, etc - 32x32 RGB images.

To see more examples with many other datasets. Please visit our wiki.

About

EasyNN is a python package designed to provide an easy-to-use Neural Network. The package is designed to work right out of the box, while also allowing the user to customize features as they see fit.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages