Spaces:

ZiyuG
/

SAM2Point

Running on Zero

SAM2Point / sam2 /csrc /connected_comp.py

Upload 42 files

d2410ba verified about 2 years ago

171 Bytes

from torch.autograd import Function

from sam2.csrc.backend import _backend

__all__ = ['connect']

def connect(mask):
    return _backend.get_connected_componnets(mask)

	from torch.autograd import Function

	from sam2.csrc.backend import _backend

	__all__ = ['connect']


	def connect(mask):
	return _backend.get_connected_componnets(mask)