bukuroo/EdgeTAM-ONNX

EdgeTAM

ONNX models for inference with EZONNX

• Model type:
  Segmentation

• Official repository:
  EdgeTAM

• Setup

  pip install git+https://github.com/ikeboo/ezonnx.git
  

• Usage

  from ezonnx import EdgeTAM, visualize_images
  tam = EdgeTAM()
  res = tam.set_image("images/cat.jpg")
  # add point with positive or not and its label
  res = tam.set_point((240, 180), True, 1)
  res = tam.set_point((50,50), True, 2)
  res = tam.set_point((400,50), True, 2)
  res = tam.set_point((150, 330), True, 3) 
  visualize_images("Masked Image",res.visualized_img)