transfer_learning_vision_classifiers 0.2.0

A simple package for creating computer vision classifiers based on transfer learning.

transfer_learning_vision_classifiers

Package for creating classifiers for computer vision, based on transfer learning.

Installation

pip install transfer_learning_vision_classifiers

Usage example

from transfer_learning_vision_classifiers import classifiers
classifier1 = classifiers.TransferLearningVisionClassifier("effnet_b2", 
                                                                ["cat", "dog"],
                                                                False)