coco_torch#
Module for loading a sample of the COCO dataset and the yolov5s model.
Functions
|
Get the COCO128 dataset and return a dataloader. |
|
Load the yolov5s (version 6.1) model and return it. |
Classes
An instance of PyTorch VisionData the represents the COCO128 dataset. |