Docker Hub images

PyTorch (using our wai.pytorchimageclass Python library)

• PyTorch 1.6.0, CUDA 10.1

  waikatodatamining/pytorch-image-classification:1.6

• PyTorch 1.6.0, CPU

  waikatodatamining/pytorch-image-classification:1.6_cpu

Tensorflow (using our wai.tfimageclass Python library)

• Tensorflow 1.14, CUDA 10.0

  waikatodatamining/tf_image_classification:1.14

• Tensorflow 1.14, CPU

  waikatodatamining/tf_image_classification:1.14_cpu

Tensorflow (using our wai.tflite_model_maker Python library)

• 2.4.3, CUDA 11.0

  waikatodatamining/tflite_model_maker:2.4.3

• 2.4.3, CPU

  waikatodatamining/tflite_model_maker:2.4.3_cpu

PyTorch/MMDetection

• MMDetection 2020-03-01, PyTorch 1.3 and CUDA 10.1

  waikatodatamining/mmdetection:2020-03-01

• MMDetection 2.16.0, PyTorch 1.6 and CUDA 10.1

  waikatodatamining/mmdetection:2.16.0

• MMDetection 2.16.0, PyTorch 1.9 and CUDA 11.1

  waikatodatamining/mmdetection:2.16.0_cuda11.1

• MMDetection 2.16.0, PyTorch 1.9 and CPU

  waikatodatamining/mmdetection:2.16.0_cpu

• MMDetection 2.18.1, PyTorch 1.9 and CUDA 11.1

  waikatodatamining/mmdetection:2.18.1_cuda11.1

• MMDetection 2.18.1, PyTorch 1.9 and CPU

  waikatodatamining/mmdetection:2.18.1_cpu

Tensorflow

• Tensorflow 1.14, Object Detection API 2019-08-31, CUDA 10.0

  waikatodatamining/tf_object_detection:1.14.0_2019-08-31

Tensorflow (using our wai.tflite_model_maker Python library)

• 2.4.3, CUDA 11.0

  waikatodatamining/tflite_model_maker:2.4.3

• 2.4.3, CPU

  waikatodatamining/tflite_model_maker:2.4.3_cpu

PyTorch/Yolov5

• Yolov5 2022-01-21, CPU

  waikatodatamining/pytorch-yolov5:2022-01-21_cpu

• Yolov5 2022-01-21, CUDA 11.1

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

• Yolov5 2022-05-31, CPU

  waikatodatamining/pytorch-yolov5:2022-05-31_cpu

• Yolov5 2022-05-31, CUDA 11.1

  waikatodatamining/pytorch-yolov5:2022-05-31_cuda11.1

Tensorflow/Keras

• image-segmentation-keras 0.3.0, Tensorflow 1.14, CUDA 10.0

  waikatodatamining/image-segmentation-keras:1.14.0_0.3.0

PyTorch

• Segmentation Models 0.2.1, PyTorch 1.9.0, CUDA 11.1

  waikatodatamining//segmentation_models:0.2.1

• MMSegmentation 0.25.0, PyTorch 1.9 and CUDA 11.1

  waikatodatamining/mmsegmentation:0.25.0_cuda11.1

• MMSegmentation 0.25.0, PyTorch 1.9 and CPU

  waikatodatamining/mmsegmentation:0.25.0_cpu

PyTorch/Detectron2

• Detectron2 0.3, PyTorch 1.6, CUDA 10.1

  waikatodatamining/detectron2:0.3

• Detectron2 0.5, PyTorch 1.9, CUDA 11.1

  waikatodatamining/detectron2:0.5

• Detectron2 0.6, PyTorch 1.9, CUDA 11.1

  waikatodatamining/detectron2:0.6

PyTorch/YOLACT++

• YOLACT++ 2020-02-11, PyTorch 1.2, CUDA 10.0

  waikatodatamining/yolactpp:2020-02-11

Tensorflow/Object Detection API

• Tensorflow 1.14, Object Detection API 2019-08-31, CUDA 10.0

  waikatodatamining/tf_object_detection:1.14.0_2019-08-31