coco_torch#

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

CocoDataset

An instance of PyTorch VisionData the represents the COCO128 dataset.