Spaces:
Running on Zero
Running on Zero
File size: 873 Bytes
f66bbd0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | """
GazeRefine
==========
Expert gaze as a test-time prompt for training-free medical image segmentation.
Quick start
-----------
from scripts.predict_single import predict
mask = predict(image_path="image.png", fixation_csv="fixations.csv")
mask.save("output_mask.png")
"""
from .backbone import FrozenDINOv3
from .gaze import get_scanpath, load_fixation_csv, generate_gaze_heatmap
from .model import GazeRefine, knn_affinity_refinement
from .metrics import compute_metrics
from .visualize import save_prediction, overlay_heatmap, overlay_mask
__all__ = [
"FrozenDINOv3",
"get_scanpath",
"load_fixation_csv",
"generate_gaze_heatmap",
"GazeRefine",
"knn_affinity_refinement",
"compute_metrics",
"save_prediction",
"overlay_heatmap",
"overlay_mask",
]
__version__ = "0.1.0"
|