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'
|