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

PyTorch/MMClassification

• 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

PyTorch/PaddleClas

• PaddleClas 2.6.0, PyTorch 2.5.1 and CUDA 11.8

  waikatodatamining/paddleclas:2.6.0_cuda11.8

• PaddleClas 2.6.0, PyTorch 1.11.0 and CPU

  waikatodatamining/paddleclas:2.6.0_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

Tensorflow (using the make_image_classifier Python library)

• 2.9.1, CUDA 11.1

  waikatodatamining/tf_image_classification2:2.9.1_cuda11.1

• 2.9.1, CPU

  waikatodatamining/tf_image_classification2:2.9.1_cpu

PyTorch/MMDetection

• 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 (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/PaddleDetection

• PaddleDetection 2.8.0, CPU

  waikatodatamining/paddledetection:2.8.0_cpu

• PaddleDetection 2.8.0, CUDA 11.8

  waikatodatamining/paddledetection:2.8.0_cuda11.8

PyTorch/Yolov5

• Yolov5 2022-11-05, CPU

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

• Yolov5 2022-11-05, CUDA 11.1

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

PyTorch/Yolov7

• Yolov7 2022-10-08, CPU

  waikatodatamining/pytorch-yolov7:2022-10-08_cpu

• Yolov7 2022-10-08, CUDA 11.1

  waikatodatamining/pytorch-yolov7:2022-10-08_cuda11.1

PyTorch/Yolov10

• Yolov10 2024-06-23, CPU

  waikatodatamining/pytorch-yolov10:2024-06-23_cpu

• Yolov10 2024-06-23, CUDA 11.7

  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

• MMSegmentation 1.2.2, PyTorch 1.11.0 and CUDA 11.3

  waikatodatamining/mmsegmentation:1.2.2-3_cuda11.3

• MMSegmentation 1.2.2, PyTorch 1.11.0 and CPU

  waikatodatamining/mmsegmentation:1.2.2-3_cpu

• PaddleSeg 2.10.0, PyTorch 2.7.0 and CUDA 11.8

  waikatodatamining/paddleseg:2.10.0-3_cuda11.8

• PaddleSeg 2.10.0, PyTorch 2.7.0 and CPU

  waikatodatamining/paddleseg:2.10.0-3_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.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

• Yolov5 2022-11-05, CPU

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

• Yolov5 2022-11-05, CUDA 11.1

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

BitNet

• BitNet 2025-04-30, CPU

  waikatodatamining/bitnet:2025-05-30_cpu

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

• Llama2, CUDA 11.7

  waikatodatamining/pytorch-huggingface-transformers:4.31.0_cuda11.7_llama2

XTuner

• XTuner 2024-02-19, CUDA 11.7

  waikatodatamining/pytorch-xtuner:2024-02-19_cuda11.7

• XTuner 0.1.15, CUDA 11.7

  waikatodatamining/pytorch-xtuner:0.1.15_cuda11.7

• XTuner 0.1.18, CUDA 11.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

• Faster Whisper 1.0.2, CPU

  waikatodatamining/python-faster-whisper:1.0.2_cpu

Huggingface

• Huggingface 4.36.0, CUDA 11.7

  waikatodatamining/pytorch-huggingface-transformers:4.36.0_cuda11.7_classification