lazylabel-gui 1.0.8__py3-none-any.whl → 1.1.0__py3-none-any.whl

lazylabel/sam_model.py

@@ -1,70 +0,0 @@
-import os
-import cv2
-import numpy as np
-import torch
-import requests
-from tqdm import tqdm
-from segment_anything import sam_model_registry, SamPredictor
-
-
-def download_model(url, download_path):
-    """Downloads file with a progress bar."""
-    print(
-        f"SAM model not found. Downloading from Meta's GitHub repository to: {download_path}"
-    )
-    response = requests.get(url, stream=True)
-    response.raise_for_status()
-    total_size_in_bytes = int(response.headers.get("content-length", 0))
-    block_size = 1024  # 1 Kibibyte
-
-    progress_bar = tqdm(total=total_size_in_bytes, unit="iB", unit_scale=True)
-    with open(download_path, "wb") as file:
-        for data in response.iter_content(block_size):
-            progress_bar.update(len(data))
-            file.write(data)
-    progress_bar.close()
-
-    if total_size_in_bytes != 0 and progress_bar.n != total_size_in_bytes:
-        print("ERROR, something went wrong during download")
-
-
-class SamModel:
-    def __init__(self, model_type, model_filename="sam_vit_h_4b8939.pth"):
-        self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-
-        # Define model URL and local cache path
-        model_url = f"https://dl.fbaipublicfiles.com/segment_anything/{model_filename}"
-        cache_dir = os.path.join(os.path.expanduser("~"), ".cache", "lazylabel")
-        os.makedirs(cache_dir, exist_ok=True)
-        model_path = os.path.join(cache_dir, model_filename)
-
-        # Download the model if it doesn't exist
-        if not os.path.exists(model_path):
-            download_model(model_url, model_path)
-
-        print(f"Loading SAM model from {model_path}...")
-        self.model = sam_model_registry[model_type](checkpoint=model_path).to(
-            self.device
-        )
-        self.predictor = SamPredictor(self.model)
-        self.image = None
-        print("SAM model loaded successfully.")
-
-    def set_image(self, image_path):
-        self.image = cv2.imread(image_path)
-        self.image = cv2.cvtColor(self.image, cv2.COLOR_BGR2RGB)
-        self.predictor.set_image(self.image)
-
-    def predict(self, positive_points, negative_points):
-        if not positive_points:
-            return None
-
-        points = np.array(positive_points + negative_points)
-        labels = np.array([1] * len(positive_points) + [0] * len(negative_points))
-
-        masks, _, _ = self.predictor.predict(
-            point_coords=points,
-            point_labels=labels,
-            multimask_output=False,
-        )
-        return masks[0]

lazylabel_gui-1.0.8.dist-info/RECORD

@@ -1,17 +0,0 @@
-lazylabel/controls.py,sha256=kminJPdKDzrSacTaUuLYXVW_nyTOcoGV_iWlMHyOfsA,10533
-lazylabel/custom_file_system_model.py,sha256=q4axrAQPQeURVz-sFeVl4s9pyY05OE9A2Ej95Ju1rXM,2488
-lazylabel/editable_vertex.py,sha256=itGcZG5MyuctGfxjINu8IJBYFFsCGuE_YtsrfHICjiw,1115
-lazylabel/hoverable_pixelmap_item.py,sha256=kJFOp7WXiyHpNf7l73TZjiob85jgP30b5MZvu_z5L3c,728
-lazylabel/hoverable_polygon_item.py,sha256=-0l8C8PfsXtJGqvZZ2qtizxHmFwO8RCwz5UfjKpDvzY,775
-lazylabel/main.py,sha256=C-ZHVFzAbnGPRDFCBTLvn3uBMuu_qq4UUsfF8cZmdns,50862
-lazylabel/numeric_table_widget_item.py,sha256=dQUlIFu9syCxTGAHVIlmbgkI7aJ3f3wmDPBz1AGK9Bg,283
-lazylabel/photo_viewer.py,sha256=PNgm0gU2gnIqvRkrGlQugdobGsKwAi3m3X6ZF487lCo,2055
-lazylabel/reorderable_class_table.py,sha256=4c-iuSkPcmk5Aey5n2zz49O85x9TQPujKG-JLxtuBCo,2406
-lazylabel/sam_model.py,sha256=9NB51Xq1P5dIxZMBdttwwRlszlJR3U5HRs83QsPLxNE,2595
-lazylabel/utils.py,sha256=sYSCoXL27OaLgOZaUkCAhgmKZ7YfhR3Cc5F8nDIa3Ig,414
-lazylabel_gui-1.0.8.dist-info/licenses/LICENSE,sha256=kSDEIgrWAPd1u2UFGGpC9X71dhzrlzBFs8hbDlENnGE,1092
-lazylabel_gui-1.0.8.dist-info/METADATA,sha256=1JlchgipzdT-mspfXCovQ1FalX0vyg4jdxFfaK1F9eA,6292
-lazylabel_gui-1.0.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-lazylabel_gui-1.0.8.dist-info/entry_points.txt,sha256=Hd0WwEG9OPTa_ziYjiD0aRh7R6Fupt-wdQ3sspdc1mM,54
-lazylabel_gui-1.0.8.dist-info/top_level.txt,sha256=YN4uIyrpDBq1wiJaBuZLDipIzyZY0jqJOmmXiPIOUkU,10
-lazylabel_gui-1.0.8.dist-info/RECORD,,