Skip to content

GitLab

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

Update inference_segment_everything.py

parent 6afd1192
Hide 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,
...@@ -139,4 +140,4 @@ show_anns_ours(mask_efficient_sam_vits, ax[2]) ...@@ -139,4 +140,4 @@ show_anns_ours(mask_efficient_sam_vits, ax[2])
ax[2].title.set_text("EfficientSAM (VIT-small)") ax[2].title.set_text("EfficientSAM (VIT-small)")
ax[2].axis('off') ax[2].axis('off')
plt.savefig("results/segmenteverything.png", bbox_inches='tight') plt.savefig("results/segmenteverything.png", bbox_inches='tight')
\ No newline at end of file
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