Datasets ⭐
DRIVE Dataset
Load DRIVE dataset in Python fast with one line of code. Blood vessels in the retina images. Stream DRIVE while training models in PyTorch and TensorFlow.
Visualization of the Drive train dataset on the Activeloop Platform.

DRIVE Dataset

What is DRIVE Dataset?

The DRIVE (Digital Retinal Images for Vessel Extraction) dataset has been created to enable studies on retinal vessel segmentation. DRIVE dataset pictures were collected from diabetic retinopathy patients in the Netherlands. The dataset contains images from 400 diabetic patients between 25-and 90 years of age. 40 photographs have been randomly selected, with only 7 showing signs of mild early diabetic retinopathy. Automatic retinal map generation and branch point extraction were used to record time or multimedia images, and synthesize the retinal image mosaic.

Download DRIVE Dataset in Python

Instead of downloading the DRIVE dataset in Python, you can effortlessly load it in Python via our open-source package Hub with just one line of code.

Load DRIVE Dataset Training Subset in Python

import hub
ds = hub.load("hub://activeloop/drive-train")

Load DRIVE Dataset Testing Subset in Python

import hub
ds = hub.load("hub://activeloop/drive-test")

DRIVE Dataset Structure

Data Fields

For training set:
  • rgb_images: tensor containing the image.
  • manual_masks/mask: tensor containing mask.
  • masks/mask: tensor containing mask.
For the testing set:
  • rgb_images: tensor containing the image.
  • masks: tensor containing mask.

DRIVE Data Splits

  • The DRIVE dataset training set is composed of 20 images. Only a single manual segmentation of the vasculature is provided for training images.
  • The DRIVE dataset test set is composed of 20 images. There are 2 manual segmentations provided where one is used as a golden standard and the other is used for comparing computer-generated segmentation with that of manual segmentation.

How to use DRIVE Dataset with PyTorch and TensorFlow in Python

Train a model on DRIVE dataset with PyTorch in Python

Let's use Hub's built-in PyTorch one-line dataloader to connect the data to the compute:
dataloader = ds.pytorch(num_workers=0, batch_size=4, shuffle=False)

Train a model on DRIVE dataset with TensorFlow in Python

dataloader = ds.tensorflow()

DRIVE Dataset Creation

Data Collection and Normalization Information

Images were taken using a Canon CR5 non-mydriatic 3 CCD camera with a 45-degree field of view (FOV). Each image was captured at 768 x 584 pixels using 8 bits per color plane. The FOV of each image is circular and has a diameter of approximately 540 pixels. In this database, the image is cropped around the FOV. For each image, a mask image depicting the FOV is provided.

Additional Information about DRIVE Dataset

DRIVE Dataset Description

  • Repository: N/A
  • Paper: Staal, J., Abràmoff, M. D., Niemeijer, M., Viergever, M. A., & Van Ginneken, B. (2004). Ridge-based vessel segmentation in color images of the retina. IEEE transactions on medical imaging, 23(4), 501-509.
  • Point of Contact: N/A

DRIVE Dataset Curators

Joes Staal, M.D. Abramoff, M. Niemeijer, M.A. Viergever, B. van Ginneken

DRIVE Dataset Licensing Information

Hub users may have access to a variety of publicly available datasets. We do not host or distribute these datasets, vouch for their quality or fairness, or claim that you have a license to use the datasets. It is your responsibility to determine whether you have permission to use the datasets under their license. If you're a dataset owner and do not want your dataset to be included in this library, please get in touch through a GitHub issue. Thank you for your contribution to the ML community!

DRIVE Dataset Citation Information

title={Ridge-based vessel segmentation in color images of the retina},
author={Staal, Joes and Abr{\`a}moff, Michael D and Niemeijer, Meindert and Viergever, Max A and Van Ginneken, Bram},
journal={IEEE transactions on medical imaging},