Tensorflow 2.9.1 image classification Docker images available

Docker images are now available for the TensorFlow 2 image classification, using the 2.9.1 release of TensorFlow:

For training the models, the make_image_classifier Python library is employed.

A tutorial on how to use these Docker images is available as well:

https://www.data-mining.co.nz/applied-deep-learning/image_classification/tf2_make_image_classifier/

Applied Deep Learning resource available

A new resources for learning how to use our Docker images is now available (also available from the menu):

https://www.data-mining.co.nz/applied-deep-learning/

In these tutorials, you learn how to prepare the data, adapt configuration files, train a model and make predictions with a model.

At this stage, it covers the following domains:

  • image classification

  • image segmentation

  • instance segmentation

  • object detection

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:

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