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.

previous

detection

next

load_dataset