Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/datarobot/pic2vec
Lightweight Image Featurization Made Easy
https://github.com/datarobot/pic2vec
csv deep-learning feature-engineering feature-extraction image-processing keras open-source python tensorflow
Last synced: 4 days ago
JSON representation
Lightweight Image Featurization Made Easy
- Host: GitHub
- URL: https://github.com/datarobot/pic2vec
- Owner: datarobot
- License: other
- Archived: true
- Created: 2017-06-14T14:44:11.000Z (over 7 years ago)
- Default Branch: master
- Last Pushed: 2021-11-15T17:47:01.000Z (about 3 years ago)
- Last Synced: 2025-01-15T14:07:04.234Z (14 days ago)
- Topics: csv, deep-learning, feature-engineering, feature-extraction, image-processing, keras, open-source, python, tensorflow
- Language: Python
- Homepage:
- Size: 8.98 MB
- Stars: 31
- Watchers: 213
- Forks: 14
- Open Issues: 7
-
Metadata Files:
- Readme: README.md
- Changelog: HISTORY.rst
- Contributing: CONTRIBUTING.rst
- License: LICENSE
Awesome Lists containing this project
README
[](https://travis-ci.org/datarobot/pic2vec) [](https://coveralls.io/github/datarobot/pic2vec?branch=master)
Pic2Vec
================
Featurize images using a small, contained pre-trained deep learning network
* Free software: BSD license
Features
--------
This is the prototype for image features engineering. Supports Python 2.7, 3.4, 3.5, 3.6, and 3.7
``pic2vec`` is a python package that performs automated feature extraction
for image data. It supports feature engineering on new image data, and allows
traditional machine learning algorithms (such as tree-based algorithms) to
train on image data.
## Input Specification
### Data Format
``pic2vec`` works on image data represented as either:
1. A directory of image files.
2. As URL pointers contained in a CSV.
3. Or as a directory of images with a CSV containing pointers to the image files.
If no CSV is provided with the directory, it automatically generates a CSV to store the features with the appropriate images.
Each row of the CSV represents a different image, and image rows can also have columns containing other data about the images as well. Each image's featurized representation will be appended as a series of new columns at the end of the appropriate image row.
### Constraints Specification
The goal of this project was to make the featurizer as easy to use and hard to break as possible. If working properly, it should be resistant to badly-formatted data, such as missing rows or columns in the csv, image mismatches between a CSV and an image directory, and invalid image formats.
However, for the featurizer to function optimally, it prefers certain constraints:
* The CSV should have no missing columns or rows, and there should be full overlap between images in the CSV and the image directory
* If checking predictions on a separate test set (such as on Kaggle), the filesystem needs to sort filepaths consistently with the sorting of the test set labels. The order in the CSV (whether generated automatically or passed in) will be considered the canonical order for the feature vectors.
The featurizer can only process .png, .jpeg, or .bmp image files. Any other images will be left out of the featurization by being represented by zero vectors in the image batch.
## Quick Start
The following Python code shows a typical usage of `pic2vec`:
```python
from pic2vec import ImageFeaturizer
image_column_name = 'images'
my_csv = 'path/to/data.csv'
my_image_directory = 'path/to/image/directory/'
my_featurizer = ImageFeaturizer(model='xception', depth=2, autosample=True)
featurized_df = my_featurizer.featurize(image_column_name, csv_path=my_csv,
image_path=my_image_directory)
```
## Examples
To get started, see the following example:
1. [Cats vs. Dogs](examples/Cats_v_Dogs_Test_Example.ipynb): Dataset from combined directory + CSV
Examples coming soon:
2. Hot Dog, Not Hot Dog: Dataset from a CSV with URLs and no image directory
## Installation
See the [Installation Guide](docs/guides/installation.md) for details.
### Installing Keras/Tensorflow
If you run into trouble installing Keras or Tensorflow as a dependency, read the [Keras installation guide](https://keras.io/#installation) and [Tensorflow installation guide](https://www.tensorflow.org/install/) for details about installing Keras/Tensorflow on your machine.
## Using Featurizer Output With DataRobot
``pic2vec`` generates a flat CSV which is ready for supervised modeling, if the data has been labelled with a variable that
can be used as a target. The images are transformed into a set of regular columns containing numeric data.
Additionally, if unlabelled, it can be used for unsupervised learning (such as anomaly detection).
### Running tests
To run the unit tests with ``pytest``, run
```
py.test tests
```
Credits
---------
This package was created with [Cookiecutter](https://github.com/audreyr/cookiecutter) and the [audreyr/cookiecutter-pypackage](https://github.com/audreyr/cookiecutter-pypackage) project template.