Huge News!Announcing our $40M Series B led by Abstract Ventures.Learn More
Socket
Sign inDemoInstall
Socket

vl-datasets

Package Overview
Dependencies
Maintainers
1
Alerts
File Explorer

Advanced tools

Socket logo

Install Socket

Detect and block malicious and high-risk dependencies

Install

vl-datasets

Open, Clean Datasets for Computer Vision.

  • 0.0.11
  • PyPI
  • Socket score

Maintainers
1

PyPi PyPi PyPi License TestedOn


Visual Layer Logo

VL-Datasets

Open, Clean, Curated Datasets for Computer Vision


🔥 We use fastdup - a free tool to clean all datasets shared in this repo.
Explore the docs »
Report Issues · Read Blog · Get In Touch · About Us

Logo Logo Logo Logo Logo

Description

vl-datasets is a Python package that provides access to clean computer vision datasets with only 2 lines of code.

For example, to get access to the clean version of the Food-101 dataset simply run:

image

We support some of the most widely used computer vision datasets. Let us know if you have additional request to support a new dataset.

All the datasets are analyzed for issues such as:

  • Duplicates.
  • Near Duplicates.
  • Broken images.
  • Outliers.
  • Dark/Bright/Blurry images.
  • Mislabels.
  • Data Leakage.

image

Why?

Computer vision is an exciting and rapidly advancing field, with new techniques and models emerging now and then. However, to develop and evaluate these models, it's essential to have reliable and standardized datasets to work with.

Even with the recent success of generative models, data quality remains an issue that's mainly overlooked. Training models will erroneours data impacts model accuracy, incurs costs in time, storage and computational resources.

We believe that access to clean and high-quality computer vision datasets leads to accurate, non-biased, and efficient model. By providing public access to vl-datasets we hope it helps advance the field of computer vision.

Datasets & Access

vl-datasets provides a convenient way to access the cleaned version of the datasets in Python.

Alternatively, for each dataset in this repo, we provide a .csv file that lists the problematic images from the dataset.

You can use the listed images in the .csv to improve the model by re-labeling the them or just simply remove it from the dataset.

We're a startup and we'd like to offer free access to the datasets as much as we can afford to. But in doing so, we'd also need your support.

We're offering select .csv files completely free with no strings attached. For access to our complete dataset and exclusive beta features, all we ask is that you sign up to be a beta tester – it's completely free and your feedback will help shape the future of our platform.

Here is a table of widely used computer vision datasets, issues we found and a link to access the .csv file.

DatasetIssuesCSVImport Statement
Food-101
  • Duplicates - 0.233 % (235)
  • Outliers - 0.076 % (77)
  • Blur - Blur - 0.183 % (185)
  • Dark - 0.043 % (43)
  • Total - 0.535 % (540)
Download here.from vl_datasets import VLFood101
Oxford-IIIT Pet
  • Duplicates - 1.021% (75)
  • Outliers - 0.095% (7)
  • Dark - 0.054% (4)
  • Total - 1.170 % (86)
Download here.from vl_datasets import VLOxfordIIITPet
LAION-1B
  • Duplicates - WIP % (WIP)
  • Outliers - WIP % (WIP)
  • Broken - WIP % (WIP)
  • Blur - WIP % (WIP)
  • Dark - WIP % (WIP)
  • Bright - WIP % (WIP)
Request access here.WIP
ImageNet-21K
  • Duplicates - 11.853 % (1,559,120)
  • Outliers - 0.085 % (11,119)
  • Blur - 0.292 % (38,458)
  • Dark - 0.179 % (23,574)
  • Bright - 0.431 % (56,754)
  • Mislabels - 3.064 % (402,963)
  • Total - 15.904 % (2,091,988)
Request access here.WIP
ImageNet-1K
  • Duplicates - 0.520 % (6,660)
  • Outliers - 0.090 % (1,150)
  • Blur - 0.200 % (2,554)
  • Dark - 0.244 % (2,997)
  • Bright - 0.058 % (746)
  • Mislabels - 0.119 % (1,518)
  • Total - 1.221 % (15,625)
Request access here.WIP
KITTI
  • Duplicates - 15.294 % (2294)
  • Outliers - 0.107 % (16)
  • Total - 15.401 % (2310)
Request access here.WIP
DeepFashion
  • Duplicates - 5.114 % (14,772)
  • Outliers - 0.037 % (107)
  • Total
Request access here.WIP
CelebA-HQ
  • Duplicates - 1.673 % (3,389)
  • Outliers - 0.077 % (157)
  • Blur - 0.512 % (1,037)
  • Dark - 0.009 % (18)
  • Mislabels - 0.006 % (13)
  • Total - 2.277 % (4,614)
Request access here.WIP
COCO
  • Duplicates - 0.123 % (201)
  • Outliers - 0.087 % (143)
  • Blur - 0.029 % (47)
  • Dark - 0.106 % (174)
  • Bright - 0.013 % (21)
  • Total - 0.358 % (586)
Request access here.WIP

Learn more on how we clean the datasets using our profilling tool here.

Installation

Option 1 - Install vl_datasets package from PyPI:

pip install vl-datasets

Option 2 - Install the bleeding edge version on GitHub:

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

Usage

To start using vl-datasets, import the clean version of the dataset with:

from vl_datasets import VLFood101

This should import the clean version of the Food101 dataset.

Next, you can load the dataset as a PyTorch Dataset.

train_dataset = VLFood101('./', split='train')
valid_dataset = VLFood101('./', split='test')

If you have a custom .csv file you can optionally pass in the file:

train_dataset = VLFood101('./', split='train', exclude_csv='my-file.csv')

The filenames listed in the .csv will be excluded in the dataset.

Next, you can load the train and validation datasets in a PyTorch training loop.

See the Learn from Examples section to learn more.

NOTE: Sign up here for free to be our beta testers and get full access to the all the .csv files for the dataset listed in this repo.

With the dataset loaded you can train a model using PyTorch training loop.

Learn from Examples

  • Dataset: VLFood101
  • Framework: PyTorch.
  • Description: Load a dataset and train a PyTorch model.
  • Dataset: VLOxfordIIITPet
  • Framework: fast.ai.
  • Description: Finetune a pretrained TIMM model using fastai.

License

vl-datasets 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.

Usage Tracking

This repository incorporates usage tracking using Sentry.io to monitor and collect valuable information about the usage of the application.

Usage tracking allows us to gain insights into how the application is being used in real-world scenarios. It provides us with valuable information that helps in understanding user behavior, identifying potential issues, and making informed decisions to improve the application.

We DO NOT collect folder names, user names, image names, image content and other personaly identifiable information.

What data is tracked?

  • Errors and Exceptions: Sentry captures errors and exceptions that occur in the application, providing detailed stack traces and relevant information to help diagnose and fix issues.
  • Performance Metrics: Sentry collects performance metrics, such as response times, latency, and resource usage, enabling us to monitor and optimize the application's performance.

To opt out, define an environment variable named SENTRY_OPT_OUT.

On Linux run the following:

export SENTRY_OPT_OUT=True

Read more on Sentry's official webpage.

Getting Help

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

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.

Keywords

FAQs


Did you know?

Socket

Socket for GitHub automatically highlights issues in each pull request and monitors the health of all your open source dependencies. Discover the contents of your packages and block harmful activity before you install or update your dependencies.

Install

Related posts

SocketSocket SOC 2 Logo

Product

  • Package Alerts
  • Integrations
  • Docs
  • Pricing
  • FAQ
  • Roadmap
  • Changelog

Packages

npm

Stay in touch

Get open source security insights delivered straight into your inbox.


  • Terms
  • Privacy
  • Security

Made with ⚡️ by Socket Inc