coco

Module for loading a sample of the COCO dataset and the yolov5s model.

Functions

load_dataset([train, batch_size, ...])	Get the COCO128 dataset and return a dataloader.
load_model([pretrained, device])	Load the yolov5s (version 6.1) model and return it.

Classes

COCOData	Class for loading the COCO dataset, inherits from DetectionData.
CocoDataset	An instance of PyTorch VisionData the represents the COCO128 dataset.

previous

detection

next

load_dataset