from segment_anything import sam_model_registry
import torch
import cv2
import numpy as np
from utils import select_random_points, save_masks

# โหลดโมเดล SAM-Med2D
sam = sam_model_registry("vit_b", "pretrain_model/sam-med2d_b.pth")
sam.eval()

# โหลดภาพที่ต้องการแบ่งส่วน
image = cv2.imread("path_to_image.png")
image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

# เลือกจุดที่ต้องการแบ่งส่วน
point = select_random_points(image_rgb)

# ประมวลผลและแบ่งส่วนภาพ
masks, scores = sam.predict(image_rgb, point)

# บันทึกมาสก์ที่ได้
save_masks(masks)