mitk::SegmentAnythingPythonService Class Reference

Segment Anything Model Python process handler class. More...

#include <mitkSegmentAnythingPythonService.h>

Inheritance diagram for mitk::SegmentAnythingPythonService:

Collaboration diagram for mitk::SegmentAnythingPythonService:

Public Types
enum	Status { READY, OFF, KILLED, CUDAError }

Public Member Functions
	mitkClassMacroItkParent (SegmentAnythingPythonService, itk::Object)
	SegmentAnythingPythonService (std::string workingDir, std::string modelType, std::string checkPointPath, unsigned int gpuId, std::string backend)
	Construct a new Segment Anything Python Service object. Specify working directory, ViT model type, checkpoint path, gpu id and backend. More...
	~SegmentAnythingPythonService ()
	Destroy the Segment Anything Python Service object. Stop the async python process and deletes temporary directories. More...
virtual void	SetMitkTempDir (std::string _arg)
virtual std::string	GetMitkTempDir () const
	mitkNewMacro5Param (SegmentAnythingPythonService, std::string, std::string, std::string, unsigned int, std::string)
void	StartAsyncProcess ()
	Creates temp directories and calls start_python_daemon function async. More...
void	StopAsyncProcess ()
	Writes KILL to the control file to stop the daemon process. More...
void	TransferImageToProcess (const Image *inputAtTimeStep, std::string &UId)
	Writes image as nifity file with unique id (UId) as file name. More...
void	TransferPointsToProcess (const std::string &triggerCSV) const
	Writes csv stringstream of points to a csv file for python daemon to read. More...
LabelSetImage::Pointer	RetrieveImageFromProcess (long timeOut=-1) const
	Waits for output nifity file from the daemon to appear and reads it as a mitk::Image. More...

Static Public Member Functions
static void	onPythonProcessEvent (itk::Object *, const itk::EventObject &, void *)
	Static function to print out everything from itk::EventObject. Used as callback in mitk::ProcessExecutor object. More...
static bool	CheckStatus ()
	Checks CurrentStatus enum variable and returns true if daemon is READY (to read files) state, false is OFF state or throws exception if daemon is found KILL or Cuda error state. More...

Static Public Attributes
static Status	CurrentStatus

Detailed Description

Segment Anything Model Python process handler class.

Definition at line 31 of file mitkSegmentAnythingPythonService.h.

Member Enumeration Documentation

Status

enum mitk::SegmentAnythingPythonService::Status

Enumerator
READY
OFF
KILLED
CUDAError

Definition at line 34 of file mitkSegmentAnythingPythonService.h.

Constructor & Destructor Documentation

SegmentAnythingPythonService()

mitk::SegmentAnythingPythonService::SegmentAnythingPythonService	(	std::string	workingDir,
		std::string	modelType,
		std::string	checkPointPath,
		unsigned int	gpuId,
		std::string	backend
	)

Construct a new Segment Anything Python Service object. Specify working directory, ViT model type, checkpoint path, gpu id and backend.

Parameters

workingDir	of python process
modelType	of ViT
checkPointPath	of specified model type
gpuId
backend	SAM or MedSAM

~SegmentAnythingPythonService()

mitk::SegmentAnythingPythonService::~SegmentAnythingPythonService

(

)

Destroy the Segment Anything Python Service object. Stop the async python process and deletes temporary directories.

Member Function Documentation

CheckStatus()

static bool mitk::SegmentAnythingPythonService::CheckStatus ( )

static

Checks CurrentStatus enum variable and returns true if daemon is READY (to read files) state, false is OFF state or throws exception if daemon is found KILL or Cuda error state.

Returns

bool

GetMitkTempDir()

virtual std::string mitk::SegmentAnythingPythonService::GetMitkTempDir ( ) const

virtual

mitkClassMacroItkParent()

mitk::SegmentAnythingPythonService::mitkClassMacroItkParent	(	SegmentAnythingPythonService	,
		itk::Object
	)

mitkNewMacro5Param()

mitk::SegmentAnythingPythonService::mitkNewMacro5Param	(	SegmentAnythingPythonService	,
		std::string	,
		std::string	,
		std::string	,
		unsigned int	,
		std::string
	)

onPythonProcessEvent()

static void mitk::SegmentAnythingPythonService::onPythonProcessEvent ( itk::Object *  , const itk::EventObject &  , void *    )

static

Static function to print out everything from itk::EventObject. Used as callback in mitk::ProcessExecutor object.

RetrieveImageFromProcess()

LabelSetImage::Pointer mitk::SegmentAnythingPythonService::RetrieveImageFromProcess

(

long

timeOut = -1

)

const

Waits for output nifity file from the daemon to appear and reads it as a mitk::Image.

Returns

LabelSetImage::Pointer

SetMitkTempDir()

virtual void mitk::SegmentAnythingPythonService::SetMitkTempDir ( std::string  _arg )

virtual

StartAsyncProcess()

void mitk::SegmentAnythingPythonService::StartAsyncProcess

(

)

Creates temp directories and calls start_python_daemon function async.

StopAsyncProcess()

void mitk::SegmentAnythingPythonService::StopAsyncProcess

(

)

Writes KILL to the control file to stop the daemon process.

TransferImageToProcess()

void mitk::SegmentAnythingPythonService::TransferImageToProcess	(	const Image *	inputAtTimeStep,
		std::string &	UId
	)

Writes image as nifity file with unique id (UId) as file name.

TransferPointsToProcess()

void mitk::SegmentAnythingPythonService::TransferPointsToProcess

(

const std::string &

triggerCSV

)

const

Writes csv stringstream of points to a csv file for python daemon to read.

Member Data Documentation

CurrentStatus

Status mitk::SegmentAnythingPythonService::CurrentStatus

static

Definition at line 115 of file mitkSegmentAnythingPythonService.h.

---

The documentation for this class was generated from the following file:

• mitkSegmentAnythingPythonService.h