The following images are available from Docker Hub:
Click on one of the domain buttons to see the relevant frameworks and images:
-
PyTorch (using our wai.pytorchimageclass Python library)
-
waikatodatamining/pytorch-image-classification:1.6
-
waikatodatamining/pytorch-image-classification:1.6_cpu
-
-
PyTorch/MMClassification
-
MMClassification 0.23.1, PyTorch 1.9 and CUDA 11.1
waikatodatamining/mmclassification:0.23.1_cuda11.1
-
MMClassification 0.23.1, PyTorch 1.9 and CPU
waikatodatamining/mmclassification:0.23.1_cpu
-
MMClassification 0.25.0, PyTorch 1.9 and CUDA 11.1
waikatodatamining/mmclassification:0.25.0_cuda11.1
-
MMClassification 0.25.0, PyTorch 1.9 and CPU
waikatodatamining/mmclassification:0.25.0_cpu
-
-
PyTorch/MMPretrain (successor of MMClassification)
-
MMPretrain 1.2.0, PyTorch 1.9 and CUDA 11.1
waikatodatamining/mmpretrain:1.2.0_cuda11.1
-
MMPretrain 1.2.0, PyTorch 1.11 and CPU
waikatodatamining/mmpretrain:1.2.0_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
-
-
Tensorflow (using the make_image_classifier Python library)
-
waikatodatamining/tf_image_classification2:2.9.1_cuda11.1
-
waikatodatamining/tf_image_classification2:2.9.1_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
-
MMDetection 2.24.1, PyTorch 1.9 and CUDA 11.1
waikatodatamining/mmdetection:2.24.1_cuda11.1
-
MMDetection 2.24.1, PyTorch 1.9 and CPU
waikatodatamining/mmdetection:2.24.1_cpu
-
MMDetection 2.27.0, PyTorch 1.9 and CUDA 11.1
waikatodatamining/mmdetection:2.27.0_cuda11.1
-
MMDetection 2.27.0, PyTorch 1.9 and CPU
waikatodatamining/mmdetection:2.27.0_cpu
-
MMDetection 3.1.0, PyTorch 1.12 and CUDA 11.3
waikatodatamining/mmdetection:3.1.0_cuda11.3
-
MMDetection 3.1.0, PyTorch 1.12 and CPU
waikatodatamining/mmdetection:3.1.0_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
-
waikatodatamining/pytorch-yolov5:2022-09-29_cpu
-
waikatodatamining/pytorch-yolov5:2022-09-29_cuda11.1
-
waikatodatamining/pytorch-yolov5:2022-11-05_cpu
-
waikatodatamining/pytorch-yolov5:2022-11-05_cuda11.1
-
-
PyTorch/Yolov7
-
waikatodatamining/pytorch-yolov7:2022-10-08_cpu
-
waikatodatamining/pytorch-yolov7:2022-10-08_cuda11.1
-
-
PyTorch/Yolov10
-
waikatodatamining/pytorch-yolov10:2024-06-23_cpu
-
waikatodatamining/pytorch-yolov10:2024-06-23_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
-
image-segmentation-keras 0.3.0, Tensorflow 2.4.1, CUDA 11.0
waikatodatamining/image-segmentation-keras:2.4.1_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
-
MMSegmentation 0.30.0, PyTorch 1.9 and CUDA 11.1
waikatodatamining/mmsegmentation:0.30.0_cuda11.1
-
MMSegmentation 0.30.0, PyTorch 1.9 and CPU
waikatodatamining/mmsegmentation:0.30.0_cpu
-
MMSegmentation 1.1.0, PyTorch 1.9 and CUDA 11.1
waikatodatamining/mmsegmentation:1.1.0_cuda11.1
-
MMSegmentation 1.1.0, PyTorch 1.9 and CPU
waikatodatamining/mmsegmentation:1.1.0_cpu
-
DEXTR 0.1.2, PyTorch 1.10.0 and CUDA 11.1
waikatodatamining/pytorch-dextr:0.1.2_cuda11.1
-
DEXTR 0.1.2, PyTorch 1.9.1 and CPU
waikatodatamining/pytorch-dextr:0.1.2_cpu
-
Segment-Anything Model 2023-04-16, PyTorch 1.13.0 and CUDA 11.6
waikatodatamining/pytorch-sam:2023-04-16_cuda11.6
-
Segment-Anything Model 2023-04-16, PyTorch 1.9.1 and CPU
waikatodatamining/pytorch-sam:2023-04-16_cpu
-
Segment-Anything in High Quality 2023-08-17, PyTorch 1.13.0 and CUDA 11.6
waikatodatamining/pytorch-sam-hq:2023-08-17_cuda11.6
-
Segment-Anything in High Quality 2023-08-17, PyTorch 1.9.1 and CPU
waikatodatamining/pytorch-sam-hq:2023-08-17_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
-
-
PyTorch/Yolov5
-
waikatodatamining/pytorch-yolov5:2022-11-05_cpu
-
waikatodatamining/pytorch-yolov5:2022-11-05_cuda11.1
-
-
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
-
-
Falcontune
-
Falcontune 20230618, CUDA 11.7
waikatodatamining/pytorch-huggingface-transformers:4.31.0_cuda11.7_falcontune_20230618
-
-
finetune-gpt2xl
-
finetune-gpt2xl 20220924, CUDA 11.1
waikatodatamining/pytorch-huggingface-transformers:4.7.0_cuda11.1_finetune-gpt2xl_20220924
-
-
Llama-2
-
waikatodatamining/pytorch-huggingface-transformers:4.31.0_cuda11.7_llama2
-
-
XTuner
-
waikatodatamining/pytorch-xtuner:2024-02-19_cuda11.7
-
waikatodatamining/pytorch-xtuner:0.1.15_cuda11.7
-
waikatodatamining/pytorch-xtuner:0.1.18_cuda11.7
-
-
OpenAI Whisper
-
Faster Whisper 1.0.2, CUDA 12.1
waikatodatamining/python-faster-whisper:1.0.2_cuda12.1
-
waikatodatamining/python-faster-whisper:1.0.2_cpu
-
-
Huggingface
-
waikatodatamining/pytorch-huggingface-transformers:4.36.0_cuda11.7_classification
-
For some of the frameworks, we have tutorials available on how to use them in practice. This includes data preparation, training and making predictions. Check out our Applied Deep Learning site for more details.