Docker Hub images
The following images are available from Docker Hub:
Image classification
-
PyTorch (using our wai.pytorchimageclass Python library)
-
waikatodatamining/pytorch-image-classification:1.6
-
waikatodatamining/pytorch-image-classification:1.6_cpu
-
-
Tensorflow (using our wai.tfimageclass Python library)
-
waikatodatamining/tf_image_classification:1.14
-
waikatodatamining/tf_image_classification:1.14_cpu
-
-
Tensorflow (using our wai.tflite_model_maker Python library)
-
waikatodatamining/tflite_model_maker:2.4.3
-
waikatodatamining/tflite_model_maker:2.4.3_cpu
-
Object detection
-
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)
-
waikatodatamining/tflite_model_maker:2.4.3
-
waikatodatamining/tflite_model_maker:2.4.3_cpu
-
-
PyTorch/Yolov5
-
waikatodatamining/pytorch-yolov5:2022-01-21_cpu
-
waikatodatamining/pytorch-yolov5:2022-01-21_cuda11.1
-
waikatodatamining/pytorch-yolov5:2022-05-31_cpu
-
waikatodatamining/pytorch-yolov5:2022-05-31_cuda11.1
-
Image segmentation
-
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
-
Instance segmentation
-
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
-