This commit is contained in:
ShangGaoG
2023-04-14 17:14:43 +08:00
parent 02cf0c17f4
commit cddb447e3f

View File

@@ -52,7 +52,7 @@ class SamControler():
return: mask, logit, painted image(mask+point) return: mask, logit, painted image(mask+point)
''' '''
# self.sam_controler.set_image(image) # self.sam_controler.set_image(image)
mask, logit = None, None
if logits is None: if logits is None:
prompts = { prompts = {
'point_coords': points, 'point_coords': points,