DEV Community

geekjr
geekjr

Posted on

QuickAI is a Python library that makes it extremely easy to experiment with state-of-the-art Machine Learning models.

GitHub: https://github.com/geekjr/quickai

As an example to perform image classification, you only need to write 2 lines of code instead of many you would otherwise have to write without quickai:

from quickai import ImageClassification
ImageClassification(<MODELARCH>, <PATH>, <SAVENAME>)
Enter fullscreen mode Exit fullscreen mode

Where:
MODELARCH is the architecture you want to use
PATH is the path to your training data
SAVENAME is the path to save your model to

There are a few more examples in the examples folder of the repo.

Discussion (0)