Deep Learning Image Embeddings/Features That Work
You are looking for generic image features for
Image classification
Image retrieval
Image similarity and so on.
Sometimes, you are not looking for latest and greatest. You just need something that just works. With image_features, you can extract such deep learning based features from images in a single line of code:
from image_features import image_features
features = image_features(['your_image_1.png', 'your_image_2.jpg'])
You can use these features to train a scikit-learn classification model:
from sklearn import linear_model
from image_features import image_features
X_train = image_features(['your_image_1.png', 'your_image_2.jpg'])
y_train = ['cat', 'dog']
clf = linear_model.LogisticRegression()
clf.fit(X_train,