Yolov5 Docker images available

Docker images are now available for Yolov5, a family of object detection architectures and models pretrained on the COCO dataset. Yolov5 performs very fast bounding box predictions.

The code used by the docker images is available from here:


The tags for the images are as follows:

  • In-house registry: public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-yolov5:2022-01-21_cuda11.1

  • Docker hub: waikatodatamining/pytorch-yolov5:2022-01-21_cuda11.1