"vscode:/vscode.git/clone" did not exist on "c83ec3555dc1399789f8157475cb5fef7e691575"
Commit 58b90e88 authored by dcuai's avatar dcuai
Browse files

Update inference_segment_everything.py

parent 6afd1192
......@@ -6,6 +6,7 @@ from PIL import Image
import io
import cv2
GRID_SIZE = 32
torch.backends.cudnn.enabled = False
from segment_anything.utils.amg import (
batched_mask_to_box,
calculate_stability_score,
......@@ -139,4 +140,4 @@ show_anns_ours(mask_efficient_sam_vits, ax[2])
ax[2].title.set_text("EfficientSAM (VIT-small)")
ax[2].axis('off')
plt.savefig("results/segmenteverything.png", bbox_inches='tight')
\ No newline at end of file
plt.savefig("results/segmenteverything.png", bbox_inches='tight')
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