models/preprocess.py

from typing import Any, Dict, List
from fastsam import FastSAM, FastSAMPrompt
import numpy as np
from configs.config import DEVICE, FAST_SAM_CONF, FAST_SAM_IMGSZ, FAST_SAM_IOU, FAST_SAM_RETINA_MASKS

def preprocess(points_data: List[Dict]) -> Any:
    
    input_points = []
    input_labels = []
    
    for point in points_data:
        input_points.append([int(point['x_']), int(point['y_'])])
        input_labels.append(int(point['flag_']))
    
    return input_points, input_labels