COCO - Common Object in Context deep learning model

Dataset

COCO dataset limited to person class.

Subset	Quantity
Training	2000
Validation	250
Testing	250

Test set is a part of the original train set, as there is no access to original test set labels.

Getting Started

Download the Dataset

Run the initialization script to download and prepare the COCO dataset:

python src/init.py

This script will download the COCO dataset (limited to the person class) and organize it into training, validation, and testing subsets.

Explore Data with FiftyOne

To visualize and explore the dataset using the FiftyOne GUI:

python -m fiftyone app launch dataset_name

Or, to use FiftyOne within a Jupyter notebook:

import fiftyone as fo

# Load your dataset
dataset = fo.load_dataset("dataset_name")

# Launch the app
session = fo.launch_app(dataset)

References

• Kaggle dataset overview
• FiftyOne documentation
• Ultralytics documentation