hujiecpp commited on
Commit
6cbef76
·
1 Parent(s): bea11e6

init project

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -256,7 +256,7 @@ def slerp_multiple(vectors, t_values):
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,7 +313,7 @@ def get_mask_from_img_sam1(yolov8, mobilesamv2, sam1_image, yolov8_image, origin
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'
 
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'