Yolov5 Docker images available

Applied Machine Learning Group, University of Waikato

2022-01-25 14:03

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:

github.com/waikato-datamining/pytorch/tree/master/yolov5

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