coralnet-toolbox 0.0.75__py2.py3-none-any.whl → 0.0.76__py2.py3-none-any.whl

coralnet_toolbox/{AutoDistill → Transformers}/QtDeployModel.py

@@ -18,6 +18,7 @@ from PyQt5.QtWidgets import (QApplication, QComboBox, QDialog,
 from coralnet_toolbox.QtProgressBar import ProgressBar
 
 from coralnet_toolbox.Results import ResultsProcessor
+from coralnet_toolbox.Results import ConvertResults
 from coralnet_toolbox.Results import MapResults
 
 from coralnet_toolbox.utilities import rasterio_open
@@ -33,13 +34,13 @@ from coralnet_toolbox.Icons import get_icon
 
 class DeployModelDialog(QDialog):
     """
-    Dialog for deploying and managing AutoDistill models.
+    Dialog for deploying and managing Transformers models.
     Allows users to load, configure, and deactivate models, as well as make predictions on images.
     """
 
     def __init__(self, main_window, parent=None):
         """
-        Initialize the AutoDistillDeployModelDialog.
+        Initialize the TransformersDeployModelDialog.
 
         Args:
             main_window: The main application window.
@@ -52,7 +53,7 @@ class DeployModelDialog(QDialog):
         self.annotation_window = main_window.annotation_window
 
         self.setWindowIcon(get_icon("coral.png"))
-        self.setWindowTitle("AutoDistill Deploy Model (Ctrl + 6)")
+        self.setWindowTitle("Transformers Deploy Model (Ctrl + 6)")
         self.resize(400, 325)
 
         # Initialize variables
@@ -66,6 +67,8 @@ class DeployModelDialog(QDialog):
         self.ontology = None
         self.class_mapping = {}
         self.ontology_pairs = []
+        
+        self.task = 'detect'
 
         # Create the layout
         self.layout = QVBoxLayout(self)
@@ -422,8 +425,6 @@ class DeployModelDialog(QDialog):
         progress_bar.close()
         # Restore cursor
         QApplication.restoreOverrideCursor()
-        # Exit the dialog box
-        self.accept()
 
     def load_new_model(self, model_name):
         """
@@ -433,8 +434,17 @@ class DeployModelDialog(QDialog):
             model_name: Name of the model to load.
             uncertainty_thresh: Threshold for uncertainty.
         """
+        
+        # Clear the model
+        self.loaded_model = None
+        self.model_name = None
+        
+        # Clear cache
+        gc.collect()
+        torch.cuda.empty_cache()
+        
         if "GroundingDINO" in model_name:
-            from coralnet_toolbox.AutoDistill.Models.GroundingDINO import GroundingDINOModel
+            from coralnet_toolbox.Transformers.Models.GroundingDINO import GroundingDINOModel
 
             model = model_name.split("-")[1].strip()
             self.model_name = model_name
@@ -443,14 +453,14 @@ class DeployModelDialog(QDialog):
                                                    device=self.main_window.device)
 
         elif "OmDetTurbo" in model_name:
-            from coralnet_toolbox.AutoDistill.Models.OmDetTurbo import OmDetTurboModel
+            from coralnet_toolbox.Transformers.Models.OmDetTurbo import OmDetTurboModel
 
             self.model_name = model_name
             self.loaded_model = OmDetTurboModel(ontology=self.ontology,
                                                 device=self.main_window.device)
 
         elif "OWLViT" in model_name:
-            from coralnet_toolbox.AutoDistill.Models.OWLViT import OWLViTModel
+            from coralnet_toolbox.Transformers.Models.OWLViT import OWLViTModel
 
             self.model_name = model_name
             self.loaded_model = OWLViTModel(ontology=self.ontology,
@@ -495,7 +505,6 @@ class DeployModelDialog(QDialog):
                     continue
 
                 results = self._apply_model(inputs)
-                results = self._update_results(results_processor, results, inputs, image_path)
                 results = self._apply_sam(results, image_path)
                 self._process_results(results_processor, results, image_path)
 
@@ -553,13 +562,6 @@ class DeployModelDialog(QDialog):
 
         return results_list
 
-    def _update_results(self, results_processor, results, inputs, image_path):
-        """Update the results to match Ultralytics format."""
-        return [results_processor.from_supervision(results,
-                                                   inputs,
-                                                   image_path,
-                                                   self.class_mapping)]
-
     def _apply_sam(self, results_list, image_path):
         """Apply SAM to the results if needed."""
         # Check if SAM model is deployed and loaded

coralnet_toolbox/{AutoDistill → Transformers}/__init__.py

@@ -1,4 +1,4 @@
-# coralnet_toolbox/AutoDistill/__init__.py
+# coralnet_toolbox/Transformers/__init__.py
 
 from .QtDeployModel import DeployModelDialog
 from .QtBatchInference import BatchInferenceDialog

coralnet_toolbox/__init__.py

@@ -1,6 +1,6 @@
 """Top-level package for CoralNet-Toolbox."""
 
-__version__ = "0.0.75"
+__version__ = "0.0.76"
 __author__ = "Jordan Pierce"
 __email__ = "jordan.pierce@noaa.gov"
 __credits__ = "National Center for Coastal and Ocean Sciences (NCCOS)"

coralnet_toolbox/utilities.py

@@ -30,21 +30,6 @@ from coralnet_toolbox.QtProgressBar import ProgressBar
 # ----------------------------------------------------------------------------------------------------------------------
 
 
-def get_available_device():
-    """
-    Get available devices
-
-    :return:
-    """
-    devices = ['cpu',]
-    if torch.cuda.is_available():
-        for i in range(torch.cuda.device_count()):
-            devices.append(f'cuda:{i}')
-    if torch.backends.mps.is_available():
-        devices.append('mps')
-    return devices
-
-
 @lru_cache(maxsize=32)
 def rasterio_open(image_path):
     """

{coralnet_toolbox-0.0.75.dist-info → coralnet_toolbox-0.0.76.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: coralnet-toolbox
-Version: 0.0.75
+Version: 0.0.76
 Summary: Tools for annotating and developing ML models for benthic imagery
 Author-email: Jordan Pierce <jordan.pierce@noaa.gov>
 License: MIT License
@@ -16,7 +16,7 @@ License-File: LICENSE.txt
 Requires-Dist: PyQt5>=5.15.11
 Requires-Dist: pyqtdarktheme
 Requires-Dist: pyqtgraph
-Requires-Dist: ultralytics>=8.3.152
+Requires-Dist: ultralytics>=8.3.191
 Requires-Dist: lap>=0.5.12
 Requires-Dist: open-clip-torch>=2.20.0
 Requires-Dist: supervision>=0.24.0
@@ -27,7 +27,7 @@ Requires-Dist: pycocotools
 Requires-Dist: ujson
 Requires-Dist: timm==0.9.2
 Requires-Dist: autodistill
-Requires-Dist: transformers>=4.55.4
+Requires-Dist: transformers>=4.56.0
 Requires-Dist: hf_xet
 Requires-Dist: x-segment-anything>=0.0.8
 Requires-Dist: yolo-tiling>=0.0.19
@@ -39,6 +39,8 @@ Requires-Dist: beautifulsoup4>=4.12.2
 Requires-Dist: webdriver_manager
 Requires-Dist: dill
 Requires-Dist: seaborn
+Requires-Dist: GPUtil
+Requires-Dist: psutil
 Provides-Extra: all
 Requires-Dist: coralnet-toolbox[extra]; extra == "all"
 Dynamic: license-file
@@ -114,6 +116,7 @@ For a complete installation guide (including CUDA setup), see the [Installation
 | **Overview** | Get the big picture | [📋 Read More](https://jordan-pierce.github.io/CoralNet-Toolbox/overview) |
 | **Installation** | Detailed setup instructions | [⚙️ Setup Guide](https://jordan-pierce.github.io/CoralNet-Toolbox/installation) |
 | **Usage** | Learn the tools | [🛠️ User Manual](https://jordan-pierce.github.io/CoralNet-Toolbox/usage) |
+| **Hot Keys** | Keyboard shortcuts | [⌨️ Shortcuts](https://jordan-pierce.github.io/CoralNet-Toolbox/hot-keys) |
 | **Classification** | Community tutorial | [🧠 AI Tutorial](https://jordan-pierce.github.io/CoralNet-Toolbox/classify) |
 
 </div>
@@ -179,7 +182,7 @@ The toolbox integrates state-of-the-art models for efficient annotation workflow
 | **Framework** | **Models** | **Capability** |
 |:---:|:---:|:---:|
 | **YOLOE** | See Anything | Visual prompt detection |
-| **AutoDistill** | Grounding DINO • OWLViT • OmDetTurbo | Zero-shot detection |
+| **Transformers** | Grounding DINO • OWLViT • OmDetTurbo | Zero-shot detection |
 
 </div>
 
@@ -277,12 +280,9 @@ uv pip install coralnet-toolbox
 ### 🚀 **Step 3: GPU Acceleration (Optional)**
 For CUDA-enabled systems:
 ```bash
-# Example for CUDA 11.8
-conda install nvidia/label/cuda-11.8.0::cuda-nvcc -y
-conda install nvidia/label/cuda-11.8.0::cuda-toolkit -y
-
+# Example for CUDA 12.9
 # Install PyTorch with CUDA support
-uv pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118 --upgrade
+uv pip install torch torchvision --index-url https://download.pytorch.org/whl/cu129 --upgrade
 ```
 
 ### 🏃‍♂️ **Step 4: Launch**

{coralnet_toolbox-0.0.75.dist-info → coralnet_toolbox-0.0.76.dist-info}/RECORD

@@ -1,30 +1,23 @@
 coralnet_toolbox/QtAnnotationWindow.py,sha256=YpE4e5gu0EcTqtzRNUcFqZl5SIjDaPK_FWxtFavfm4k,40950
 coralnet_toolbox/QtConfidenceWindow.py,sha256=L5hR23uW91GpqnsNS9R1XF3zCTe2aU7w0iDoQMV0oyE,16190
-coralnet_toolbox/QtEventFilter.py,sha256=qL3-FzQ2iZX1yITJb2mjn9_Q09nNNqKsgx-oCz4aT9Y,7717
+coralnet_toolbox/QtEventFilter.py,sha256=utGSwtKlKl-3qepN6aaVQm4sL513bfuJpA3-3VTqnDU,7701
 coralnet_toolbox/QtImageWindow.py,sha256=B_38VR_0QARyo05UEvInmQKd3N74Dfb0g_QDeGCTUuY,51291
 coralnet_toolbox/QtLabelWindow.py,sha256=O-mLtE6ycuuGloYYZX0Z9JYZtsBMmspqNeJWslrjfFc,51419
-coralnet_toolbox/QtMainWindow.py,sha256=ctwSZiPql-nafxNacifHdZtNnIY-u-xvhKWptC7aBTk,122154
+coralnet_toolbox/QtMainWindow.py,sha256=4Znd0fezebQ-cWOKHjHTPVfj_DPH45A7tnUfxUEiqOQ,123615
 coralnet_toolbox/QtPatchSampling.py,sha256=Ehj06auBGfQwIruLNYQjF8eFOCpl8G72p42UXXb2mUo,29013
-coralnet_toolbox/QtProgressBar.py,sha256=kFcq0ftj4Hljesev-jwWMLVc_OOjlOwQrGc4r2yGMAs,7877
+coralnet_toolbox/QtProgressBar.py,sha256=L9V1dD9NQ5K-IK2NhJtESoieWGd1ULLsACEXuUDE4Ck,7922
+coralnet_toolbox/QtSystemMonitor.py,sha256=KSsUZ11GcHPH0Q7Pl8TUsFSB_jnWzNqGmtnf7IYYXpQ,16567
 coralnet_toolbox/QtWorkArea.py,sha256=YXRvHQKpWUtWyv_o9lZ8rmxfm28dUOG9pmMUeimDhQ4,13578
-coralnet_toolbox/__init__.py,sha256=kdpxYZD8oO9MggK9OdvUGRRyKAyDN3KovZN_ORG88jM,207
+coralnet_toolbox/__init__.py,sha256=VV5fKKSOZQ99ek8daqcG9QSiYWU3u7JvjF5L9yq8tfA,207
 coralnet_toolbox/main.py,sha256=6j2B_1reC_KDmqvq1C0fB-UeSEm8eeJOozp2f4XXMLQ,1573
-coralnet_toolbox/utilities.py,sha256=xF13RAadniOPQEzzE4L9-iYrL4G_acMwuMBCp8_8vV8,31498
+coralnet_toolbox/utilities.py,sha256=Up6_z0n-8p7KccFLgBvlSnhfgX8B_mVTSmIfajpkkug,31173
 coralnet_toolbox/Annotations/QtAnnotation.py,sha256=4KxqDe_WPMGK18OYHZ1NqWzV87ARv8MnLAQdHFXo-Yg,27527
 coralnet_toolbox/Annotations/QtMaskAnnotation.py,sha256=Hs8p-Lxv4OYqcx4Y7dayer1KxTaX-G0kuQe-W2JuGIE,16269
 coralnet_toolbox/Annotations/QtMultiPolygonAnnotation.py,sha256=u0vPQPeaBCEdw5lMFygpBbjDKDSjvtV7Jkj03MDfj8M,16463
 coralnet_toolbox/Annotations/QtPatchAnnotation.py,sha256=huoCsPloSA5uCZxG7pwIoO02GcHFDwW8-VohLxbCVnY,16608
-coralnet_toolbox/Annotations/QtPolygonAnnotation.py,sha256=D8VajJ1pBtb5NOSYIspAxonkiVWn0yuWSNo8vRjreSw,33371
-coralnet_toolbox/Annotations/QtRectangleAnnotation.py,sha256=FnKDHRNg_v9pUEqhdyvbGBGSxtNww3AZU_rulYU0878,21357
+coralnet_toolbox/Annotations/QtPolygonAnnotation.py,sha256=NYzca916tfZDOxbP9TBC5CaFEoiMiZec8QkwHtpfGLQ,35198
+coralnet_toolbox/Annotations/QtRectangleAnnotation.py,sha256=F49Cc3MyPPHQp-qhfjsFACE5ZRwT4Hsq0jDi8nt9iVg,22492
 coralnet_toolbox/Annotations/__init__.py,sha256=bpMldC70tT_lzMrOdBNDkEhG9dCX3tXEBd48IrcUg3E,419
-coralnet_toolbox/AutoDistill/QtBatchInference.py,sha256=k871aW3XRX8kc4BDaS1aipbPh9WOZxgmilF2c4KOdVA,5646
-coralnet_toolbox/AutoDistill/QtDeployModel.py,sha256=Fycm7wuydUfr1E2CUy00ridiI2JaNDZqAeoVB_HVydY,25923
-coralnet_toolbox/AutoDistill/__init__.py,sha256=-cJSCr3HSVcybbkvdSZY_zz9EDLESq9A3gisHu3gIgM,206
-coralnet_toolbox/AutoDistill/Models/GroundingDINO.py,sha256=xG20nLOrKjtzRhZznIIdwFXxBJ7RCeQ7h1z0V0J6trE,2781
-coralnet_toolbox/AutoDistill/Models/OWLViT.py,sha256=disVxSQ80sS4SVYdwrQocFP_LN6YDQQhzfeORWe4veU,2572
-coralnet_toolbox/AutoDistill/Models/OmDetTurbo.py,sha256=i2k9C0U8CzojKvv58CE_4wvquvR_JHUHRCe93Yzb5QQ,2526
-coralnet_toolbox/AutoDistill/Models/QtBase.py,sha256=P9dzGgzOZJZr-hQltAIswWqUyfaUP40GcXc_X11GOv8,4220
-coralnet_toolbox/AutoDistill/Models/__init__.py,sha256=3woEIkWjoNLlZhNijnyAwAimsBoy2AGCt_tks3Y4q6M,259
 coralnet_toolbox/BreakTime/QtBreakout.py,sha256=KYlhLMHF_5HVkjR8JDjbNu8CB6SHsEpECAywXqWVw10,54763
 coralnet_toolbox/BreakTime/QtSnake.py,sha256=XxmV64A_1avYf1uC_fXQpOZV3kCetz3CqboQsFwSIJk,22398
 coralnet_toolbox/BreakTime/__init__.py,sha256=7d_CMXp7T872NV-a6xaGU4oq5wjWAWGyrnd-YD3BDJo,150
@@ -42,13 +35,13 @@ coralnet_toolbox/Explorer/QtFeatureStore.py,sha256=3VwGezs1stmu65Z4ZQpvY27rGEIJq
 coralnet_toolbox/Explorer/QtSettingsWidgets.py,sha256=wwgMje5hga6GpJsAwFYXqd3G_Hew0oKmtu3tEr061Hs,36042
 coralnet_toolbox/Explorer/QtViewers.py,sha256=2qtzxSQNsRoHYUdm2t046QCUJd51pKOts5DNtgMRmUY,69408
 coralnet_toolbox/Explorer/__init__.py,sha256=wZPhf2oaUUyIQ2WK48Aj-4q1ENIZG2dGl1HF_mjhI6w,116
-coralnet_toolbox/Explorer/transformer_models.py,sha256=JlNNht2xEXlVC258-AXE4S_HBh37LGkEloNxFeSEamQ,2063
+coralnet_toolbox/Explorer/transformer_models.py,sha256=yNgGoYEIWcSgCnCOSKdAatG4M6GRbtf4pVmUWGPxvWo,2878
 coralnet_toolbox/Explorer/yolo_models.py,sha256=GicZrypDE699gut5KEW68Ui_KiTk4Ojt1uRkyDWJVI8,3473
 coralnet_toolbox/IO/QtExportAnnotations.py,sha256=xeaS0BukC3cpkBIGT9DXRqHmvHhp-vOU47h6EoANpNg,4474
 coralnet_toolbox/IO/QtExportCoralNetAnnotations.py,sha256=4royhF63EmeOlSIBX389EUjjvE-SF44_maW6qm52mdA,2778
 coralnet_toolbox/IO/QtExportGeoJSONAnnotations.py,sha256=9HkHjQTRtH4VnYa50c5pyqQz30R_6gIH5i3xFF6kDWI,27759
 coralnet_toolbox/IO/QtExportLabels.py,sha256=Vsav0wd1EK4g065aEWvxyNuvvM9BFB7UXxz6IJzwVBU,2588
-coralnet_toolbox/IO/QtExportMaskAnnotations.py,sha256=U-2sQyA154La5-QCovkJkrz_2-2bmuZytTw05VKinCE,24268
+coralnet_toolbox/IO/QtExportMaskAnnotations.py,sha256=7prXGsFFXjF1dvaybEqmwBO-N1th0c-s8GsedmyB_RQ,32651
 coralnet_toolbox/IO/QtExportTagLabAnnotations.py,sha256=JL4r1a6_PUjCzWQjMxOzxtkF2gyqIttpD14OxEcW-dA,11330
 coralnet_toolbox/IO/QtExportTagLabLabels.py,sha256=e6OL8UNtLRAJrovfs1cxVz0k2bHuJXdVmO-A0OVpgSk,3164
 coralnet_toolbox/IO/QtExportViscoreAnnotations.py,sha256=AUTzVB-N9uwlQPSds74YXyPVZzEHph7HDq01R88OBJY,19166
@@ -93,6 +86,7 @@ coralnet_toolbox/Icons/rocket.png,sha256=iMlRGlrNBS_dNBD2XIpN4RSrphCGbw_Ds1AYJ01
 coralnet_toolbox/Icons/select.png,sha256=twnMIO9ylQYjvyGnAR28V6K3ds6xpArZQTrvf0uxS6g,1896
 coralnet_toolbox/Icons/settings.png,sha256=rklROt3oKrfEk_qwN9J-JwvKok08iOkZy3OD4oNsLJQ,1376
 coralnet_toolbox/Icons/snake.png,sha256=cwcekSkXwDi_fhtTU48u7FN4bIybbY53cWK0n7-IN9A,2361
+coralnet_toolbox/Icons/system_monitor.png,sha256=ui6377kyFMHLnbfSFiE5NAJVnC16tku4RDi7Rv5vJ-0,739
 coralnet_toolbox/Icons/target.png,sha256=jzb-S_sXWT8MfbvefhDNsuTdAZgV2nGf1ieawaCkByM,1702
 coralnet_toolbox/Icons/tile.png,sha256=WiXKBpWVBfPv7gC8dnkc_gW3wuLQmLUyxYMWEM-G9ZU,382
 coralnet_toolbox/Icons/transparent.png,sha256=ZkuGkVzh6zLVNau1Wj166-TtUlbCRqJObGt4vxMxnLk,1098
@@ -183,21 +177,21 @@ coralnet_toolbox/Rasters/RasterManager.py,sha256=GuTeuty3x4k8DWZ34FQ4BTNwRfjWu0l
 coralnet_toolbox/Rasters/RasterTableModel.py,sha256=8ebirBkTUSy5Rdsoq10sqzDQBoYCH_Hu40dPiUhtlzc,15311
 coralnet_toolbox/Rasters/__init__.py,sha256=Pi88uDQbtWxwHfJFdlsvbkwGNhtlyM_013l8bbJlFfw,428
 coralnet_toolbox/Results/CombineResults.py,sha256=QrHyKhMrjNDtQ98PQabUflHhyv_8KXTGqU30tw9amV8,4523
-coralnet_toolbox/Results/ConvertResults.py,sha256=uh7tfJxDiH_cJHxWsr8KQQOcsfzC59-k6PQDVtl3ZaY,6015
+coralnet_toolbox/Results/ConvertResults.py,sha256=f9L1C7JQMVt7x2eSG4CAqpUDRhhc6eQZvTaqCiGUWmk,6310
 coralnet_toolbox/Results/MapResults.py,sha256=sDieekB93RVKvD-9mW1zRsHOs85shYVIQklt9-kFJZI,8088
 coralnet_toolbox/Results/Masks.py,sha256=C-zoobRaWXP_QdGcL7ZgSxytHOBdHIBUbQuGnoMZthE,5183
-coralnet_toolbox/Results/ResultsProcessor.py,sha256=q_UZNYggpZyY4_P6RpDLbY1ygNH49GXlP1s9ZFI3yF0,17261
+coralnet_toolbox/Results/ResultsProcessor.py,sha256=WPyq4nETrnk9WpMaeFw0rlQRFqZkzAYz_qttT3_jaoA,17332
 coralnet_toolbox/Results/__init__.py,sha256=WPdlq8aXzjrdQo5T3UqFh7jxge33iNEHiSRAmm0eJuw,630
 coralnet_toolbox/SAM/QtBatchInference.py,sha256=UyuYLfPF4JrOmmuMOzshbKDEEribV669d9LURmuu6gg,6866
-coralnet_toolbox/SAM/QtDeployGenerator.py,sha256=S05ZdoimMWQV0imLTymGFDEhpdQqd2kCdlHKKZ_gX2w,26498
-coralnet_toolbox/SAM/QtDeployPredictor.py,sha256=4Lwp4e95HX3rdL25jfEuA0EH6mP_2u30zDLuikJfhgo,23944
+coralnet_toolbox/SAM/QtDeployGenerator.py,sha256=UtJH1ZQ9g_Wqa4s09PBMCrmNRpqUZZaTMpE7lfH_Xq0,26491
+coralnet_toolbox/SAM/QtDeployPredictor.py,sha256=7jOFFnNqY7Ylr1IKjTI_YSrPYgCxHpI4ZBda7Kp806g,24413
 coralnet_toolbox/SAM/__init__.py,sha256=Zxd75pFMrt5DfSmNNVSsQeCucIQ2rVaEiS0hT_OVIMM,293
 coralnet_toolbox/SeeAnything/QtBatchInference.py,sha256=k3aftVzva84yATB4Su5DSI0lhkHDggUg3mVAx4AHmjw,7134
-coralnet_toolbox/SeeAnything/QtDeployGenerator.py,sha256=CFGhrMOkP9YGhjqRCrdsTlQR5qk7k4OaSpeQsg4M2cs,68456
-coralnet_toolbox/SeeAnything/QtDeployPredictor.py,sha256=HGZLgoqYwPiwMMc7vZLlGcaOpHDmjU-C3dPs4v3duPk,26610
+coralnet_toolbox/SeeAnything/QtDeployGenerator.py,sha256=5wG2XKhHzdNXjmG3u5Npc7TGBfFLMWPHzYQ12NlZvd0,68578
+coralnet_toolbox/SeeAnything/QtDeployPredictor.py,sha256=sfaaJoDFM2ntdqD0CXsTSHRZIuCRfviVg4vqhG0sGdI,26804
 coralnet_toolbox/SeeAnything/QtTrainModel.py,sha256=dQ6ZkeIr1migU-edGO-gQMENVP4o7WJsIANlSVhFK8k,28031
 coralnet_toolbox/SeeAnything/__init__.py,sha256=4OgG9-aQ6_RZ942-Ift_q-kkp14kObMT4lDIIx9YSxQ,366
-coralnet_toolbox/Tile/QtTileBatchInference.py,sha256=chSo-TOwpnAgIQpTetZnbhWQIBzbDJO01xAQ3z0-ubk,10730
+coralnet_toolbox/Tile/QtTileBatchInference.py,sha256=oo7NeqEwwmxMEGBMo5v2SCnyQwJYq9pg-9yj3G_-vaM,10727
 coralnet_toolbox/Tile/QtTileCreation.py,sha256=Cw6q0ZVXx0hU4uMuXA3OZ3_5bKu6oQgnlMFwaXSZbS0,39963
 coralnet_toolbox/Tile/__init__.py,sha256=BlV-1bO9u-olfNAIvukUMPRzlw8dx-ayjrjMRQ-bSsk,463
 coralnet_toolbox/Tile/TileDataset/QtBase.py,sha256=5zWtnNVbril0i3aVFv5q0VyMYcFP0Qb09ylqQ4smBXw,18623
@@ -212,7 +206,7 @@ coralnet_toolbox/Tools/QtPatchTool.py,sha256=57vFeR2jQ_VQRlMEIC_mH8NigUqOlVvmhaV
 coralnet_toolbox/Tools/QtPolygonTool.py,sha256=mQB2NuGGUmo7W4Pf8DJI9PDa3PW7Hu2nOMf4uTyWAq8,11236
 coralnet_toolbox/Tools/QtRectangleTool.py,sha256=VJWKktDiDN1TUZATRSe_1kjppFjV0AMYDZ8bK78dCXc,8672
 coralnet_toolbox/Tools/QtResizeSubTool.py,sha256=cWJDx8PEtxoCLQKuyEyZ6ccBzFKau9j1djrSSpDgaq8,6524
-coralnet_toolbox/Tools/QtSAMTool.py,sha256=PKSQU0gBsbI5rDCHRs46fUsS8QDkaqYT7aF8CEreNBQ,34068
+coralnet_toolbox/Tools/QtSAMTool.py,sha256=c719q573SRDxr5mLxNB7K5YNvZnWQYcFxzdm_I4f08Y,36529
 coralnet_toolbox/Tools/QtSeeAnythingTool.py,sha256=ZjqaM0At7MCB5R0wgibph75FkH2ZIePfRnXQUQko6wE,38515
 coralnet_toolbox/Tools/QtSelectSubTool.py,sha256=_FIjLhnEVY19Q87jhRKXGdghNfMBxxy_sECAIUo0BZA,3294
 coralnet_toolbox/Tools/QtSelectTool.py,sha256=qAXRIGmjdzWjaH6GwhvlQSodZuFa6OnyckzNVfVDG2w,20983
@@ -222,9 +216,17 @@ coralnet_toolbox/Tools/QtTool.py,sha256=2MCjT151gYBN8KbsK0GX4WOrEg1uw3oeSkp7Elw1
 coralnet_toolbox/Tools/QtWorkAreaTool.py,sha256=ApsIiEbkyWFWKW6qnFPPnL_Wgs2xa_Edif5kZU5_n8M,22733
 coralnet_toolbox/Tools/QtZoomTool.py,sha256=F9CAoABv1jxcUS7dyIh1FYjgjOXYRI1xtBPNIR1g62o,4041
 coralnet_toolbox/Tools/__init__.py,sha256=UYStZw1eA_yJ07IVli1MYSvk0pSCs1aS169LcQo630s,867
-coralnet_toolbox-0.0.75.dist-info/licenses/LICENSE.txt,sha256=AURacZ_G_PZKqqPQ9VB9Sqegblk67RNgWSGAYKwXXMY,521
-coralnet_toolbox-0.0.75.dist-info/METADATA,sha256=PRpW31atpfxGpf3hPm1eQdqdGEg5OlY6WEXZWFhFWr4,15328
-coralnet_toolbox-0.0.75.dist-info/WHEEL,sha256=JNWh1Fm1UdwIQV075glCn4MVuCRs0sotJIq-J6rbxCU,109
-coralnet_toolbox-0.0.75.dist-info/entry_points.txt,sha256=oEeMoDlJ_2lq95quOeDHIx9hZpubUlSo80OLtgbcrbM,63
-coralnet_toolbox-0.0.75.dist-info/top_level.txt,sha256=SMWPh4_9JfB8zVpPOOvjucV2_B_hvWW7bNWmMjG0LsY,17
-coralnet_toolbox-0.0.75.dist-info/RECORD,,
+coralnet_toolbox/Transformers/QtBatchInference.py,sha256=Adry1H-oIMV6Ppo8yRJRx79oeG1yUthT5jqszj7EJ20,5764
+coralnet_toolbox/Transformers/QtDeployModel.py,sha256=oH82XtP07f0n6fUhlPADUbFujTJt0ppSvitX15KeuaQ,25669
+coralnet_toolbox/Transformers/__init__.py,sha256=Oef7mKgwlK_hi5ZtXlRTvpyKhf98JPwBELWE3zjrp9U,207
+coralnet_toolbox/Transformers/Models/GroundingDINO.py,sha256=V77tapTLsXtTISbqsV9ZSGYgkJQTI1RKsT95QagYhqk,2747
+coralnet_toolbox/Transformers/Models/OWLViT.py,sha256=l9R9XKN7grw6gF7EC9DWxF5sUsApLfi0WO-zj6pVVHU,2781
+coralnet_toolbox/Transformers/Models/OmDetTurbo.py,sha256=vaXaQNqBCvnEFcPMt6x_EJI-gf-Wy3eYZPak7a527WY,2592
+coralnet_toolbox/Transformers/Models/QtBase.py,sha256=AYGTOxopOXYrHSDWAenvyxAAgqHbQs3zIox-c4BX9YQ,4533
+coralnet_toolbox/Transformers/Models/__init__.py,sha256=icJnQkt2vZksubEJuih0sT0q2vLR_Y-12WuTGquvxt8,260
+coralnet_toolbox-0.0.76.dist-info/licenses/LICENSE.txt,sha256=AURacZ_G_PZKqqPQ9VB9Sqegblk67RNgWSGAYKwXXMY,521
+coralnet_toolbox-0.0.76.dist-info/METADATA,sha256=MlZwkwAUGZWKcCzQVqIpHkuoenuz2arqnEIzviPsFps,15381
+coralnet_toolbox-0.0.76.dist-info/WHEEL,sha256=JNWh1Fm1UdwIQV075glCn4MVuCRs0sotJIq-J6rbxCU,109
+coralnet_toolbox-0.0.76.dist-info/entry_points.txt,sha256=oEeMoDlJ_2lq95quOeDHIx9hZpubUlSo80OLtgbcrbM,63
+coralnet_toolbox-0.0.76.dist-info/top_level.txt,sha256=SMWPh4_9JfB8zVpPOOvjucV2_B_hvWW7bNWmMjG0LsY,17
+coralnet_toolbox-0.0.76.dist-info/RECORD,,

coralnet_toolbox/AutoDistill/Models/GroundingDINO.py

@@ -1,81 +0,0 @@
-from dataclasses import dataclass
-
-import cv2
-import numpy as np
-
-import supervision as sv
-
-from transformers import AutoProcessor, AutoModelForZeroShotObjectDetection
-
-from autodistill.detection import CaptionOntology
-from autodistill.helpers import load_image
-
-from coralnet_toolbox.AutoDistill.Models.QtBase import QtBaseModel
-
-
-# ----------------------------------------------------------------------------------------------------------------------
-# Classes
-# ----------------------------------------------------------------------------------------------------------------------
-
-
-@dataclass
-class GroundingDINOModel(QtBaseModel):
-    def __init__(self, ontology: CaptionOntology, model="SwinB", device: str = "cpu"):
-        super().__init__(ontology, device)
-        
-        if model == "SwinB":
-            model_name = "IDEA-Research/grounding-dino-base"
-        else:
-            model_name = "IDEA-Research/grounding-dino-tiny"
-            
-        self.processor = AutoProcessor.from_pretrained(model_name, use_fast=True)
-        self.model = AutoModelForZeroShotObjectDetection.from_pretrained(model_name).to(self.device)
-
-    def _process_predictions(self, image, texts, class_idx_mapper, confidence):
-        """Process model predictions for a single image."""
-        inputs = self.processor(text=texts, images=image, return_tensors="pt").to(self.device)
-        outputs = self.model(**inputs)
-
-        results = self.processor.post_process_grounded_object_detection(
-            outputs,
-            inputs.input_ids,
-            box_threshold=confidence,
-            text_threshold=confidence,
-            target_sizes=[image.shape[:2]],
-        )[0]
-
-        boxes, scores, labels = (
-            results["boxes"],
-            results["scores"],
-            results["text_labels"],
-        )
-
-        final_boxes, final_scores, final_labels = [], [], []
-
-        for box, score, label in zip(boxes, scores, labels):
-            try:
-                box = box.detach().cpu().numpy().astype(int).tolist()
-                score = score.item()
-                # Grounding Dino issues...
-                label = class_idx_mapper[label.split(" ")[0]]
-                
-                # Amplify scores
-                if score < confidence:
-                    continue
-
-                final_boxes.append(box)
-                final_scores.append(score)
-                final_labels.append(label)
-            
-            except Exception as e:
-                print(f"Error: Issue converting predictions:\n{e}")
-                continue
-
-        if len(final_boxes) == 0:
-            return None
-
-        return sv.Detections(
-            xyxy=np.array(final_boxes),
-            class_id=np.array(final_labels),
-            confidence=np.array(final_scores)
-        )

coralnet_toolbox/AutoDistill/Models/OWLViT.py

@@ -1,76 +0,0 @@
-from dataclasses import dataclass
-
-import cv2
-import numpy as np
-
-import supervision as sv
-
-from transformers import OwlViTForObjectDetection, OwlViTProcessor
-
-from autodistill.detection import CaptionOntology
-from autodistill.helpers import load_image
-
-from coralnet_toolbox.AutoDistill.Models.QtBase import QtBaseModel
-
-
-# ----------------------------------------------------------------------------------------------------------------------
-# Classes
-# ----------------------------------------------------------------------------------------------------------------------
-
-
-@dataclass
-class OWLViTModel(QtBaseModel):
-    def __init__(self, ontology: CaptionOntology, device: str = "cpu"):
-        super().__init__(ontology, device)
-        
-        model_name = "google/owlvit-base-patch32"
-        self.processor = OwlViTProcessor.from_pretrained(model_name, use_fast=True)
-        self.model = OwlViTForObjectDetection.from_pretrained(model_name).to(self.device)
-
-    def _process_predictions(self, image, texts, class_idx_mapper, confidence):
-        """Process model predictions for a single image."""
-        inputs = self.processor(text=texts, images=image, return_tensors="pt").to(self.device)
-        outputs = self.model(**inputs)
-
-        results = self.processor.post_process_object_detection(
-            outputs,
-            threshold=confidence,
-            target_sizes=[image.shape[:2]]
-        )[0]
-
-        boxes, scores, labels = (
-            results["boxes"],
-            results["scores"],
-            results["labels"],
-        )
-
-        final_boxes, final_scores, final_labels = [], [], []
-
-        for box, score, label in zip(boxes, scores, labels):
-            try:
-                box = box.detach().cpu().numpy().astype(int).tolist()
-                score = score.item()
-                label_index = label.item()
-                class_label = texts[label_index]
-                label = class_idx_mapper[class_label]
-                
-                # Filter by confidence
-                if score < confidence:
-                    continue
-
-                final_boxes.append(box)
-                final_scores.append(score)
-                final_labels.append(label)
-            
-            except Exception as e:
-                print(f"Error: Issue converting predictions:\n{e}")
-                continue
-
-        if len(final_boxes) == 0:
-            return None
-
-        return sv.Detections(
-            xyxy=np.array(final_boxes),
-            class_id=np.array(final_labels),
-            confidence=np.array(final_scores)
-        )

coralnet_toolbox/AutoDistill/Models/OmDetTurbo.py

@@ -1,75 +0,0 @@
-from dataclasses import dataclass
-
-import cv2
-import numpy as np
-
-import supervision as sv
-
-from transformers import AutoProcessor, OmDetTurboForObjectDetection
-
-from autodistill.detection import CaptionOntology
-from autodistill.helpers import load_image
-
-from coralnet_toolbox.AutoDistill.Models.QtBase import QtBaseModel
-
-
-# ----------------------------------------------------------------------------------------------------------------------
-# Classes
-# ----------------------------------------------------------------------------------------------------------------------
-
-
-@dataclass
-class OmDetTurboModel(QtBaseModel):
-    def __init__(self, ontology: CaptionOntology, device: str = "cpu"):
-        super().__init__(ontology, device)
-        
-        model_name = "omlab/omdet-turbo-swin-tiny-hf"
-        self.processor = AutoProcessor.from_pretrained(model_name, use_fast=True)
-        self.model = OmDetTurboForObjectDetection.from_pretrained(model_name).to(self.device)
-
-    def _process_predictions(self, image, texts, class_idx_mapper, confidence):
-        """Process model predictions for a single image."""
-        inputs = self.processor(text=texts, images=image, return_tensors="pt").to(self.device)
-        outputs = self.model(**inputs)
-
-        results = self.processor.post_process_grounded_object_detection(
-            outputs,
-            threshold=confidence,
-            target_sizes=[image.shape[:2]],
-            text_labels=texts,
-        )[0]
-
-        boxes, scores, labels = (
-            results["boxes"],
-            results["scores"],
-            results["text_labels"],
-        )
-
-        final_boxes, final_scores, final_labels = [], [], []
-
-        for box, score, label in zip(boxes, scores, labels):
-            try:
-                box = box.detach().cpu().numpy().astype(int).tolist()
-                score = score.item()
-                label = class_idx_mapper[label]
-                
-                # Amplify scores
-                if score < confidence:
-                    continue
-
-                final_boxes.append(box)
-                final_scores.append(score)
-                final_labels.append(label)
-            
-            except Exception as e:
-                print(f"Error: Issue converting predictions:\n{e}")
-                continue
-
-        if len(final_boxes) == 0:
-            return None
-
-        return sv.Detections(
-            xyxy=np.array(final_boxes),
-            class_id=np.array(final_labels),
-            confidence=np.array(final_scores)
-        )