Segment-Anything Model Docker images available

Applied Machine Learning Group, University of Waikato

2023-05-01 13:16

Docker images for Segment-Anything Model (SAM) are now available.

SAM is a great tool for aiding a human annotating images for image segmentation or object detection, as it can determine a relatively good outline of an object based on either a point or a box. Only pre-trained models are available.

The code used by the docker images is available from here:

github.com/waikato-datamining/pytorch/tree/master/segment-anything

The tags for the images are as follows:

In-house registry:
- public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-sam:2023-04-16_cuda11.6
- public.aml-repo.cms.waikato.ac.nz:443/pytorch/pytorch-sam:2023-04-16_cpu
Docker hub:
- waikatodatamining/pytorch-sam:2023-04-16_cuda11.6
- waikatodatamining/pytorch-sam:2023-04-16_cpu