Skip to content

GitLab

Commit 58b90e88 authored by dcuai's avatar dcuai
Browse files

Update inference_segment_everything.py

parent 6afd1192
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
inference_segment_everything.py
View file @ 58b90e88
...@@ -6,6 +6,7 @@ from PIL import Image ...@@ -6,6 +6,7 @@ from PIL import Image
import io import io
import cv2 import cv2
GRID_SIZE = 32 GRID_SIZE = 32
torch.backends.cudnn.enabled = False
from segment_anything.utils.amg import ( from segment_anything.utils.amg import (
batched_mask_to_box, batched_mask_to_box,
calculate_stability_score, calculate_stability_score,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment