Commit
•
33402bc
1
Parent(s):
971a795
Update app.py
Browse files
app.py
CHANGED
@@ -65,7 +65,6 @@ def image_transform_grounding_for_vis(init_image):
|
|
65 |
model = load_model_hf(config_file, ckpt_repo_id, ckpt_filenmae)
|
66 |
|
67 |
def segment(image, sam_model, boxes):
|
68 |
-
sam_model.set_image(image)
|
69 |
H, W, _ = image.shape
|
70 |
boxes_xyxy = box_ops.box_cxcywh_to_xyxy(boxes) * torch.Tensor([W, H, W, H])
|
71 |
|
|
|
65 |
model = load_model_hf(config_file, ckpt_repo_id, ckpt_filenmae)
|
66 |
|
67 |
def segment(image, sam_model, boxes):
|
|
|
68 |
H, W, _ = image.shape
|
69 |
boxes_xyxy = box_ops.box_cxcywh_to_xyxy(boxes) * torch.Tensor([W, H, W, H])
|
70 |
|