init project
Browse files
app.py
CHANGED
@@ -256,7 +256,7 @@ def slerp_multiple(vectors, t_values):
|
|
256 |
|
257 |
return interpolated_vector
|
258 |
|
259 |
-
|
260 |
def get_mask_from_img_sam1(yolov8, mobilesamv2, sam1_image, yolov8_image, original_size, input_size, transform):
|
261 |
|
262 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
@@ -313,7 +313,7 @@ def get_mask_from_img_sam1(yolov8, mobilesamv2, sam1_image, yolov8_image, origin
|
|
313 |
|
314 |
return ret_mask
|
315 |
|
316 |
-
|
317 |
def get_cog_feats(images, sam2, siglip, siglip_processor, yolov8, mobilesamv2):
|
318 |
|
319 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
|
256 |
|
257 |
return interpolated_vector
|
258 |
|
259 |
+
@torch.no_grad
|
260 |
def get_mask_from_img_sam1(yolov8, mobilesamv2, sam1_image, yolov8_image, original_size, input_size, transform):
|
261 |
|
262 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
|
313 |
|
314 |
return ret_mask
|
315 |
|
316 |
+
@torch.no_grad
|
317 |
def get_cog_feats(images, sam2, siglip, siglip_processor, yolov8, mobilesamv2):
|
318 |
|
319 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|