Image classifier model that automate every steps in image classification
Project description
This library can be used for any simple image classification problems, You only have to give the data path, #of epochs, and if you have any model that you created you can give that also (if none given the model if use it's own model).
- After importing the library create a object of it by providing the path of the data and other parameters.
- Then call the create_model function with the object.
- Then the library will starts to work starting with cleaning the data by checking for any unsupported or duplicated files and deleting them.
- After cleanig it will preprocessing the images and splits it to train/test/validation, and it also plots the images according to the classes
- The if no model is provided it will create it's own model for traning.
- And Starts to train for the #of epochs given (default=70).
- You can use the funtion plot_performance() to plot the model loss and accuracy growth.
- model_evaluation() function evaluate the model based on the test data.
- And you can use the function predict() to predict any new values.
- And to save the model use model_save() function
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
imoc-0.0.1.tar.gz
(4.9 kB
view hashes)
Built Distribution
imoc-0.0.1-py3-none-any.whl
(5.2 kB
view hashes)