visuallayer

visuallayer copied to clipboard

Unleash the Full Power of Your Visual Data

Explore the docs »
Report Issues · Read Blog · Get In Touch · About Us

💫 Check out VL Datasets release blog post

Description

visuallayer SDK is an open-source Python package that offers access and extensibility to the Visual Layer platform from your code.

While the platform offers a high-level overview and visualization of your data, the SDK affords you the flexibility to integrate into your favorite machine learning frameworks and environments (e.g. Jupyter Notebook) using Python.

Installation

The easiest way to use the visuallayer SDK is to install it from PyPI. On your machine, run:

pip install visuallayer

Optionally, you can also install the bleeding edge version on GitHub by running:

pip install git+https://github.com/visual-layer/visuallayer.git@main --upgrade

VL Datasets

The visuallayer SDK also lets you access VL Datasets - a collection of clean versions of widely used computer vision datasets.

For example with only 2 lines of code, load the clean vl datasets version of the ImageNet-1k dataset with:

import visuallayer as vl
dataset = vl.datasets.zoo.load('vl-imagenet-1k')

#Export to PyTorch
train_dataset = dataset.export(output_format='pytorch', split='train')

#PyTorch training loop

Note: visuallayer does not automatically download the ImageNet dataset, you should make sure to obtain usage rights to the dataset and download it into your current working directory first.

When we say "clean" we mean that the datasets loaded by visuallayer SDK were flagged from common issues such as duplicates, mislabels, outliers, dark/bright/ blurry and data leakage. See full description for issues support in our documentation.

Dataset Zoo

We provide a Dataset Zoo where you can find all information for each VL Dataset.

For each dataset in the zoo, we ran an analysis using VL Profiler and found issues pertaining to the original dataset. The following table is a detailed breakdown of the issues for each dataset.

We provide here full details on each issues removed from a VL Dataset (a vl dataset card). The clean version of a dataset is prefixed with vl- to differentiate it from the original dataset. You can also freely download all found issues CSV.

VL Dataset Card	Original Dataset	Explore	Issues CSV	Hugging Face Dataset
vl-imagenet-21k	ImageNet-21K
vl-imagenet-1k	ImageNet-1K
vl-laion-1b	LAION-1B
vl-kitti	KITTI
vl-coco	COCO
vl-deepfashion	DeepFashion
vl-celeba-hq	CelebA-HQ
vl-places365	Places365
vl-food-101	Food-101
vl-oxford-iiit-pet	Oxford-IIIT Pet

We will continue to support more datasets. Here are a few currently in our roadmap:

• EuroSAT
• Flickr30k
• INaturalist
• SVHN
• Cityscapes
• RVL-CDIP
• DocLayNet

Let us know if you have additional request to support a specific dataset.

Note: If you'd like to use our cloud tool and discover issues with your own dataset, sign up to use our cloud platform for free.

Usage

The following sections show how to use the visuallayer SDK to load, inspect and export a VL Dataset.

Loading a dataset

We offer handy functions to load datasets from the Dataset Zoo. First, let's list the datasets in the zoo with:

import visuallayer as vl
vl.datasets.zoo.list_datasets()

which currently outputs:

['vl-oxford-iiit-pets',
 'vl-imagenet-21k',
 'vl-imagenet-1k',
 'vl-food101',
 'oxford-iiit-pets',
 'imagenet-21k',
 'imagenet-1k',
 'food101']

To load the dataset:

vl.datasets.zoo.load('vl-oxford-iiit-pets')

This loads the clean version of the Oxford IIIT Pets dataset where all of the problematic images are excluded from the dataset.

To load the original Oxford IIIT Pets dataset, simply drop the vl- prefix:

original_pets_dataset = vl.datasets.zoo.load('oxford-iiit-pets')

This loads the original dataset with no modifications.

Inspecting a dataset

Now that you have a dataset loaded, you can view information pertaining to that dataset with:

my_pets.info

This prints out high-level information about the original Dataset. In this example, we used the Pets Dataset from Oxford.

Metadata:
--> Name - vl-oxford-iiit-pets
--> Description - A modified version of the original Oxford IIIT Pets Dataset removing dataset issues.
--> License - Creative Commons Attribution-ShareAlike 4.0 International (CC BY-SA 4.0)
--> Homepage URL - https://www.robots.ox.ac.uk/~vgg/data/pets/
--> Number of Images - 7349
--> Number of Images with Issues - 109

If you'd like to view the issues related to the dataset, run:

my_pets.report

which outputs:

| Reason    | Count | Pct   |
|-----------|-------|-------|
| Duplicate | 75    | 1.016 |
| Outlier   | 7     | 0.095 |
| Dark      | 4     | 0.054 |
| Leakage   | 23    | 0.627 |
| Total     | 109   | 1.792 |

Now that you've seen the issues with the dataset, you can visualize them on screen. There are two options to visualize the dataset issues.

Option 1 - Using the Visual Layer Profiler (VL Profiler) - Provides an extensive capability to view, group, sort, and filter the dataset issues. Sign-up for free.

Here's the visualization using the VL Profiler:

Option 2 - In Jupyter Notebook - Provides a limited but convenient way to view the dataset without leaving your notebook.

To visualize the issues using Option 2 in your notebook, run:

my_pets.explore()

This should output an interactive table in your Jupyter Notebook like the following.

In the interactive table, you can view the issues, sort, filter, search, and compare the images side by side.

By default, the .explore() load the top 50 issues from the dataset covering all issue types. If you'd like a more granular control, you can change the num_images and issue arguments.

For example:

pets_dataset.explore(num_images=100, issue='Duplicate')

The interactive table provides a convenient but limited way to visualize dataset issues. For a more extensive visualization, view the issues using the Visual Layer Profiler.

Check out the documentation and blog page for the Visual Layer Profiler for more info.

Exporting a dataset

If you'd like to use a loaded dataset to train a model, you can conveniently export the dataset with:

test_dataset = my_pets.export(output_format="pytorch", split="test")

This exports the Dataset into a Pytorch Dataset object that can be used readily with a PyTorch training loop.

Alternatively, you can export the Dataset to a DataFrame with:

test_dataset = pets_dataset.export(output_format="csv", split="test")

Learn from Examples

In this section, we show an end-to-end example of how to load, inspect and export a dataset and then train using PyTorch and fastai framework.

	Dataset: vl-food101 Framework: PyTorch. Description: Load a dataset and train a PyTorch model.
	Dataset: vl-food101 Frameworks: PyTorch + Hugging Face Dataset Description: Load VL Datasets using Hugging Face Datasets and train a PyTorch model.
	Dataset: vl-oxford-iiit-pet Framework: fast.ai. Description: Finetune a pretrained TIMM model using fastai.
	Dataset: vl-imagenet-1k Framework: PyTorch. Description: Load cleaned ImageNet dataset and train a PyTorch model.

License

visuallayer SDK is licensed under the Apache 2.0 License. See LICENSE.

However, you are bound to the usage license of the original dataset. It is your responsibility to determine whether you have permission to use the dataset under the dataset's license. We provide no warranty or guarantee of accuracy or completeness.

Telemetry

export SENTRY_OPT_OUT=True

Getting Help

Get help from the Visual Layer team or community members via the following channels -

• Slack.
• GitHub issues.
• Discussion forum.

About Visual-Layer

Visual Layer is founded by the authors of XGBoost, Apache TVM & Turi Create - Danny Bickson, Carlos Guestrin and Amir Alush.

Learn more about Visual Layer here.