matchypatchy 0.2.1rc1__tar.gz

matchypatchy-0.2.1rc1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Conservation Technology Lab
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

matchypatchy-0.2.1rc1/PKG-INFO

@@ -0,0 +1,195 @@
+Metadata-Version: 2.4
+Name: matchypatchy
+Version: 0.2.1rc1
+Summary: GUI tool for human validation of AI-powered animal re-identification
+Author-email: Kyra Swanson <tswanson@sdzwa.org>
+Project-URL: Homepage, https://github.com/onservationtechlab/matchypatchy
+Project-URL: Issues, https://github.com/onservationtechlab/matchypatchy/issues
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: animl-lite==3.3.2
+Requires-Dist: chromadb>=1.3.4
+Requires-Dist: pyqt6>=6.9.0
+Dynamic: license-file
+
+# MatchyPatchy
+<br/>
+
+[User Manual](https://conservationtechlab.github.io/matchypatchy/)
+
+MatchyPatchy is an open-source GUI tool for human validation of AI-powered animal re-identification.\
+It was developed by the San Diego Zoo Wildlife Alliance Conservation Technology Lab.\
+Author: Kyra Swanson, (c) 2025
+<br/>
+
+---
+## Installation
+
+Matchypatchy is a stand-alone executable built in Python. Download it [here](). \
+[Github Repository](https://github.com/conservationtechlab/matchypatchy)
+<br/>
+
+---
+## Getting Started
+<br/>
+
+### Download Models
+In order to calculate potential matches, MatchyPatchy requires an animal object detector such as [MegaDetector]()
+and an embedding extractor such as [MiewID]().  Both models and more can be obtained from the SDZWA server by selecting 
+'Download Models' and checking the box associated with which models you want to download.
+<br/>
+
+### Select Survey
+In order to add media to the database, you must first specify a Survey using 
+the dropdown of available surveys. To add a new survey, 
+
+1. Select 'Manage Surveys'
+2. Select 'New' 
+3. Enter Survey Name
+4. Enter Survey Region
+5. (Optional) Enter Survey Start Year
+6. (Optional) Enter Survey End Year
+7. Select 'Ok'
+<br/>
+
+### 1. Import Images
+Images can be imported via spreadsheet or by manually selecting a directory of images.
+
+**From CSV**
+1. Select '1. Import from CSV'
+2. Open .csv file
+3. Select column headers that correspond to image Filepath, Timestamp, Survey, and Station
+4. Optionally select column headers that correspond to image Region, Sequence ID, External ID,
+    Viewpoint, Species, Individual ID, and Comment
+5. Select 'Ok'
+<br/>
+
+**From Folder**
+1. Select '1. Import from Folder'
+2. Open directory containing image files. Can be a whole Survey containg multiple Stations or single Station.
+3. Select the directory level that corresponds to Station name if available
+4. Select 'Ok'
+<br/>
+
+### 2. Process Images
+Run images through a sequence of AI models to obtain predicted matches.\
+Note: Models must be accessible in MatchyPatchy's "Models" folder - see Configuration for more.
+
+1. Select '2. Process'
+2. (Optional) Check Calculate Sequence (improves performance for videos and images captured in quick succession)
+3. **Select Detector Model** - recommended MegaDetector v5a
+4. (Optional) Select Species Classifier Model
+5. **Select Re-Indentification Model** - required to obtain predicted matches, recommended MiewIDv3
+6. (Optional) Select Viewpoint Model (improves performance by estimating animal viewpoint and limiting search space)
+<br/>
+
+### 3. Validate 
+After processing, this pageview allows you to review the imported media for errors, make changes, delete images, etc.
+It can be accessed at any time after image import. 
+<br/>
+
+Double-clicking on a cell will allow you to edit it.\
+Double-clicking on a single row will pull up a screen to edit multiple fields for an image.\
+Selecting one or more row (either by clicking on the row number or checking the select box) will enable the ability
+to mass edit particular fields, duplicate entries, or delete entries.
+<br/>
+
+The **"Show"** Option allows you view the full images or detected ROIS.\
+The **"Save"** Option saves all edits to the database.\
+The **"Undo"** Option undoes the last edit made.
+<br/>
+
+**Filters** \
+Images can be filtered by Region, Survey, Station, Species, and Individual.\
+You can also choose to show only "Unidentified" ROIs.\
+You can also display all images marked as "Favorite"
+<br/>
+
+### 4. Match
+After processing, potential matches can be validated at this pageview.
+The left side contains a **Query** image and the right side contains potential **Matches**.
+If "Calculate Sequence" was selected during the processing stage, the entire **Sequence** will be viewable on the Query side.
+The pageview will display the images and associated metadata. 
+<br/>
+
+_View Options_\
+Adjustments can be made to the **Maximum Number of Matches** shown for each image (within a sequence) and to the maximum distance
+threshold between the two images to qualify as a match. The **Cosine** or **L2** distance of the Match from the Query is displayed above the 
+match image. Select which distance metric to use in the dropdown. You must select **"Recalculate Matches"** if changes are made to either option
+in order to display those changes.\
+After matches are validated, you can **"Query by Individual"** to do quality control and ensure all images associated with
+an individual are indeed valid.\
+Images can be **Filtered** by Region, Survey, or Station to narrow the available matches shown.\
+If **"Viewpoint"** was calculated at the processing step, you can filter both the **Query** and the **Match** images by Viewpoint.\
+The **"View Data"** Button at the bottom brings you to the Media Validation pageview. 
+<br/>
+
+_Image Options_\
+You can adjust the **Brightness, Contrast** and **Sharpness** of each image, as well as **Zoom** using the mouse scroll wheel.\
+The **"Reset"** button returns the image to its original form.\
+The **"Edit Image"** option allows you to make changes to the metadata.\
+The **"Open Image"** option opens the image in your default image viewer.\
+The **"♥"** option marks the image with the "Favorite" tag.
+<br/>
+
+To confirm a match between the Query and the Match image, select the **"Match" button** in the middle, or press the 'M' or 'Space Key'.
+Pressing Match again will undo the match, asking you to confirm first.\
+You can navigate between Query Sequences and Matches by using the **"<<"** and **">>"** buttons, or by using the arrow keys.\
+You can also toggle the **Viewpoint** by pressing 'V'.
+<br/>
+
+### 5. Export
+The database can be exported as a .csv file. Simply select '5. Export' and specify the file name and location. 
+<br/>
+
+---
+## Database Management
+<br/>
+
+### Definitions
+* Survey - An ecological study employing camera traps to investigate wildlife of a particular area.
+* Region - A geographical region that may host one or more camera trap surveys.
+* Station - Reference to a physical location containing one or camera traps.
+* Species - The taxonomic and common names refering to a class of animal.
+* Media - Image (and soon video) files obtained from camera traps.
+* ROI - Regions of Interest within media files that contain animals. 
+* Individual - A particular, singular example of a given species, can be named/ID'd.
+* Sequence ID - A unique number associated with a set of sequential images or frames in a video.
+* External ID - A unique number used to reference MatchyPatchy output with other datamanagement tools.
+* Viewpoint - The side of the body of the animl facing the camera, eg "Left", "Right", "Top"
+* Comment - Notes or concerns about a particular media file
+* Favorite - A tag marking an image as noteworthy.
+<br/>
+
+### Manage Tables
+**Survey, Station,** and **Species** tables can be managed independently. 
+<br/>
+
+### CSV Import
+<br/>
+
+
+---
+## Configuration
+<br/>
+
+### Available Models
+Models currently available and compatible with MatchyPatchy:\
+_Detector Models_:
+- MegaDetector v5a
+- MegaDetector v5b
+
+_Re-ID Models_: 
+- MiewID v3
+- MiewID v2
+
+_Viewpoint Models_:
+- SDZWA Viewpoint
+<br/>
+
+### Link to New Database
+Database files can be shared across multiple users of MatchyPatchy. To link to a new .db file, 
+<br/>
+
+### Clear Data 

matchypatchy-0.2.1rc1/README.md

@@ -0,0 +1,180 @@
+# MatchyPatchy
+<br/>
+
+[User Manual](https://conservationtechlab.github.io/matchypatchy/)
+
+MatchyPatchy is an open-source GUI tool for human validation of AI-powered animal re-identification.\
+It was developed by the San Diego Zoo Wildlife Alliance Conservation Technology Lab.\
+Author: Kyra Swanson, (c) 2025
+<br/>
+
+---
+## Installation
+
+Matchypatchy is a stand-alone executable built in Python. Download it [here](). \
+[Github Repository](https://github.com/conservationtechlab/matchypatchy)
+<br/>
+
+---
+## Getting Started
+<br/>
+
+### Download Models
+In order to calculate potential matches, MatchyPatchy requires an animal object detector such as [MegaDetector]()
+and an embedding extractor such as [MiewID]().  Both models and more can be obtained from the SDZWA server by selecting 
+'Download Models' and checking the box associated with which models you want to download.
+<br/>
+
+### Select Survey
+In order to add media to the database, you must first specify a Survey using 
+the dropdown of available surveys. To add a new survey, 
+
+1. Select 'Manage Surveys'
+2. Select 'New' 
+3. Enter Survey Name
+4. Enter Survey Region
+5. (Optional) Enter Survey Start Year
+6. (Optional) Enter Survey End Year
+7. Select 'Ok'
+<br/>
+
+### 1. Import Images
+Images can be imported via spreadsheet or by manually selecting a directory of images.
+
+**From CSV**
+1. Select '1. Import from CSV'
+2. Open .csv file
+3. Select column headers that correspond to image Filepath, Timestamp, Survey, and Station
+4. Optionally select column headers that correspond to image Region, Sequence ID, External ID,
+    Viewpoint, Species, Individual ID, and Comment
+5. Select 'Ok'
+<br/>
+
+**From Folder**
+1. Select '1. Import from Folder'
+2. Open directory containing image files. Can be a whole Survey containg multiple Stations or single Station.
+3. Select the directory level that corresponds to Station name if available
+4. Select 'Ok'
+<br/>
+
+### 2. Process Images
+Run images through a sequence of AI models to obtain predicted matches.\
+Note: Models must be accessible in MatchyPatchy's "Models" folder - see Configuration for more.
+
+1. Select '2. Process'
+2. (Optional) Check Calculate Sequence (improves performance for videos and images captured in quick succession)
+3. **Select Detector Model** - recommended MegaDetector v5a
+4. (Optional) Select Species Classifier Model
+5. **Select Re-Indentification Model** - required to obtain predicted matches, recommended MiewIDv3
+6. (Optional) Select Viewpoint Model (improves performance by estimating animal viewpoint and limiting search space)
+<br/>
+
+### 3. Validate 
+After processing, this pageview allows you to review the imported media for errors, make changes, delete images, etc.
+It can be accessed at any time after image import. 
+<br/>
+
+Double-clicking on a cell will allow you to edit it.\
+Double-clicking on a single row will pull up a screen to edit multiple fields for an image.\
+Selecting one or more row (either by clicking on the row number or checking the select box) will enable the ability
+to mass edit particular fields, duplicate entries, or delete entries.
+<br/>
+
+The **"Show"** Option allows you view the full images or detected ROIS.\
+The **"Save"** Option saves all edits to the database.\
+The **"Undo"** Option undoes the last edit made.
+<br/>
+
+**Filters** \
+Images can be filtered by Region, Survey, Station, Species, and Individual.\
+You can also choose to show only "Unidentified" ROIs.\
+You can also display all images marked as "Favorite"
+<br/>
+
+### 4. Match
+After processing, potential matches can be validated at this pageview.
+The left side contains a **Query** image and the right side contains potential **Matches**.
+If "Calculate Sequence" was selected during the processing stage, the entire **Sequence** will be viewable on the Query side.
+The pageview will display the images and associated metadata. 
+<br/>
+
+_View Options_\
+Adjustments can be made to the **Maximum Number of Matches** shown for each image (within a sequence) and to the maximum distance
+threshold between the two images to qualify as a match. The **Cosine** or **L2** distance of the Match from the Query is displayed above the 
+match image. Select which distance metric to use in the dropdown. You must select **"Recalculate Matches"** if changes are made to either option
+in order to display those changes.\
+After matches are validated, you can **"Query by Individual"** to do quality control and ensure all images associated with
+an individual are indeed valid.\
+Images can be **Filtered** by Region, Survey, or Station to narrow the available matches shown.\
+If **"Viewpoint"** was calculated at the processing step, you can filter both the **Query** and the **Match** images by Viewpoint.\
+The **"View Data"** Button at the bottom brings you to the Media Validation pageview. 
+<br/>
+
+_Image Options_\
+You can adjust the **Brightness, Contrast** and **Sharpness** of each image, as well as **Zoom** using the mouse scroll wheel.\
+The **"Reset"** button returns the image to its original form.\
+The **"Edit Image"** option allows you to make changes to the metadata.\
+The **"Open Image"** option opens the image in your default image viewer.\
+The **"♥"** option marks the image with the "Favorite" tag.
+<br/>
+
+To confirm a match between the Query and the Match image, select the **"Match" button** in the middle, or press the 'M' or 'Space Key'.
+Pressing Match again will undo the match, asking you to confirm first.\
+You can navigate between Query Sequences and Matches by using the **"<<"** and **">>"** buttons, or by using the arrow keys.\
+You can also toggle the **Viewpoint** by pressing 'V'.
+<br/>
+
+### 5. Export
+The database can be exported as a .csv file. Simply select '5. Export' and specify the file name and location. 
+<br/>
+
+---
+## Database Management
+<br/>
+
+### Definitions
+* Survey - An ecological study employing camera traps to investigate wildlife of a particular area.
+* Region - A geographical region that may host one or more camera trap surveys.
+* Station - Reference to a physical location containing one or camera traps.
+* Species - The taxonomic and common names refering to a class of animal.
+* Media - Image (and soon video) files obtained from camera traps.
+* ROI - Regions of Interest within media files that contain animals. 
+* Individual - A particular, singular example of a given species, can be named/ID'd.
+* Sequence ID - A unique number associated with a set of sequential images or frames in a video.
+* External ID - A unique number used to reference MatchyPatchy output with other datamanagement tools.
+* Viewpoint - The side of the body of the animl facing the camera, eg "Left", "Right", "Top"
+* Comment - Notes or concerns about a particular media file
+* Favorite - A tag marking an image as noteworthy.
+<br/>
+
+### Manage Tables
+**Survey, Station,** and **Species** tables can be managed independently. 
+<br/>
+
+### CSV Import
+<br/>
+
+
+---
+## Configuration
+<br/>
+
+### Available Models
+Models currently available and compatible with MatchyPatchy:\
+_Detector Models_:
+- MegaDetector v5a
+- MegaDetector v5b
+
+_Re-ID Models_: 
+- MiewID v3
+- MiewID v2
+
+_Viewpoint Models_:
+- SDZWA Viewpoint
+<br/>
+
+### Link to New Database
+Database files can be shared across multiple users of MatchyPatchy. To link to a new .db file, 
+<br/>
+
+### Clear Data 

matchypatchy-0.2.1rc1/pyproject.toml

@@ -0,0 +1,42 @@
+[project]
+name = "matchypatchy"
+dynamic = ["version"]
+authors = [
+  { name="Kyra Swanson", email="tswanson@sdzwa.org" },
+]
+description = "GUI tool for human validation of AI-powered animal re-identification"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "animl-lite==3.3.2",
+    "chromadb>=1.3.4",
+    "pyqt6>=6.9.0",
+]
+
+[project.urls]
+Homepage = "https://github.com/onservationtechlab/matchypatchy"
+Issues = "https://github.com/onservationtechlab/matchypatchy/issues"
+
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[tool.setuptools.dynamic]
+version = {attr = "matchypatchy.__version__"}
+
+[tool.setuptools]
+packages = ["matchypatchy"]
+package-dir = {"" = "src"}
+ 
+[tool.setuptools.package-data]
+matchypatchy = ["assets/**/*", "assets/graphics/*"]
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+addopts = "-v"
+
+[tool.pylint.master]
+extension-pkg-whitelist = ["PyQt6"]
+
+[tool.pylint.messages_control]
+disable = ["no-name-in-module"]

matchypatchy-0.2.1rc1/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

matchypatchy-0.2.1rc1/src/matchypatchy/__init__.py

@@ -0,0 +1,105 @@
+__version__ = "0.2.1rc1"
+
+from matchypatchy import config
+from matchypatchy import database
+from matchypatchy import gui
+from matchypatchy import logging_config
+from matchypatchy import threads
+
+from matchypatchy.config import (asset_path, mpConfig, resource_path,)
+from matchypatchy.database import (EditObject, IMAGE_EXT, MatchyPatchyDB,
+                                   THUMBNAIL_NOTFOUND, THUMBNAIL_SIZE,
+                                   TZ_CONVERT_DICT, VIDEO_EXT,
+                                   check_missing_thumbnails, fetch_individual,
+                                   fetch_media, fetch_media_thumbnails,
+                                   fetch_regions, fetch_roi, fetch_roi_media,
+                                   fetch_roi_thumbnails,
+                                   fetch_station_names_from_id, fetch_stations,
+                                   fetch_surveys, get_frame, get_roi_bbox,
+                                   get_sequence, get_sha256,
+                                   individual_roi_dict, location, media,
+                                   media_count, mpdb, save_media_thumbnail,
+                                   save_roi_thumbnail, sequence_roi_dict,
+                                   setup, setup_chromadb, setup_database,
+                                   thumbnails,)
+from matchypatchy.gui import (AlertPopup, ClickableSlider, ComboBoxDelegate,
+                              ComboBoxSeparator, ConfigPopup, DisplayBase,
+                              DisplayCompare, DisplayMedia, FilterBar,
+                              FilterBox, HorizontalSeparator, ImageAdjustBar,
+                              ImageWidget, ImportCSVPopup, ImportFolderPopup,
+                              IndividualFillPopup, IndividualPopup,
+                              MLDownloadPopup, MLOptionsPopup, MainWindow,
+                              ManualQueryContainer, MediaEditPopup, MediaTable,
+                              MediaWidget, MetadataPanel, NoHoverDelegate,
+                              PairXPopup, QC_QueryContainer, QueryContainer,
+                              READMEPopup, SequenceSelector, SliderWithLabel,
+                              StandardButton, StationFillPopup, StationPopup,
+                              SurveyFillPopup, SurveyPopup, TextEditWithSignal,
+                              ThreePointSlider, UploadManagerPopup,
+                              VerticalSeparator, VideoPlayerBar, VideoViewer,
+                              VideoWidget, dialogs, display_base,
+                              display_compare, display_media, gui_assets,
+                              main_gui, manual_query, media_table, popup_alert,
+                              popup_config, popup_import_csv,
+                              popup_import_folder, popup_individual,
+                              popup_media_edit, popup_ml, popup_pairx,
+                              popup_readme, popup_station, popup_survey,
+                              popup_uploads, qc_query, query, widget_filterbar,
+                              widget_image_adjustment, widget_media, widgets,)
+from matchypatchy.logging_config import (get_logger, setup_logger,)
+from matchypatchy.threads import (AnimlThread, BasePathUpdateThread,
+                                  BuildManifestThread, CSVImportThread,
+                                  CSVMigrateThread, DownloadMLThread,
+                                  FavoriteMatchObject, FetchTableThread,
+                                  FolderImportThread, MEGADETECTORv1000_SIZE,
+                                  MatchEmbeddingThread, MatchObject,
+                                  ReIDThread, SequenceThread,
+                                  VerifyNewBaseDirsThread, animl_thread,
+                                  delete, get_path, import_thread,
+                                  is_valid_reid_model, load_model,
+                                  match_object, match_thread,
+                                  model_download_thread, reid_thread,
+                                  sequence_thread, table_thread,
+                                  update_model_yml,)
+
+__all__ = ['AlertPopup', 'AnimlThread', 'BasePathUpdateThread',
+           'BuildManifestThread', 'CSVImportThread', 'CSVMigrateThread',
+           'ClickableSlider', 'ComboBoxDelegate', 'ComboBoxSeparator',
+           'ConfigPopup', 'DisplayBase', 'DisplayCompare', 'DisplayMedia',
+           'DownloadMLThread', 'EditObject', 'FavoriteMatchObject',
+           'FetchTableThread', 'FilterBar', 'FilterBox', 'FolderImportThread',
+           'HorizontalSeparator', 'IMAGE_EXT', 'ImageAdjustBar', 'ImageWidget',
+           'ImportCSVPopup', 'ImportFolderPopup', 'IndividualFillPopup',
+           'IndividualPopup', 'MEGADETECTORv1000_SIZE', 'MLDownloadPopup',
+           'MLOptionsPopup', 'MainWindow', 'ManualQueryContainer',
+           'MatchEmbeddingThread', 'MatchObject', 'MatchyPatchyDB',
+           'MediaEditPopup', 'MediaTable', 'MediaWidget', 'MetadataPanel',
+           'NoHoverDelegate', 'PairXPopup', 'QC_QueryContainer',
+           'QueryContainer', 'READMEPopup', 'ReIDThread', 'SequenceSelector',
+           'SequenceThread', 'SliderWithLabel', 'StandardButton',
+           'StationFillPopup', 'StationPopup', 'SurveyFillPopup',
+           'SurveyPopup', 'THUMBNAIL_NOTFOUND', 'THUMBNAIL_SIZE',
+           'TZ_CONVERT_DICT', 'TextEditWithSignal', 'ThreePointSlider',
+           'UploadManagerPopup', 'VIDEO_EXT', 'VerifyNewBaseDirsThread',
+           'VerticalSeparator', 'VideoPlayerBar', 'VideoViewer', 'VideoWidget',
+           'animl_thread', 'asset_path', 'check_missing_thumbnails', 'config',
+           'database', 'delete', 'dialogs', 'display_base', 'display_compare',
+           'display_media', 'fetch_individual', 'fetch_media',
+           'fetch_media_thumbnails', 'fetch_regions', 'fetch_roi',
+           'fetch_roi_media', 'fetch_roi_thumbnails',
+           'fetch_station_names_from_id', 'fetch_stations', 'fetch_surveys',
+           'get_frame', 'get_logger', 'get_path', 'get_roi_bbox',
+           'get_sequence', 'get_sha256', 'gui', 'gui_assets', 'import_thread',
+           'individual_roi_dict', 'is_valid_reid_model', 'load_model',
+           'location', 'logging_config', 'main_gui', 'manual_query',
+           'match_object', 'match_thread', 'media', 'media_count',
+           'media_table', 'model_download_thread', 'mpConfig', 'mpdb',
+           'popup_alert', 'popup_config', 'popup_import_csv',
+           'popup_import_folder', 'popup_individual', 'popup_media_edit',
+           'popup_ml', 'popup_pairx', 'popup_readme', 'popup_station',
+           'popup_survey', 'popup_uploads', 'qc_query', 'query', 'reid_thread',
+           'resource_path', 'save_media_thumbnail', 'save_roi_thumbnail',
+           'sequence_roi_dict', 'sequence_thread', 'setup', 'setup_chromadb',
+           'setup_database', 'setup_logger', 'table_thread', 'threads',
+           'thumbnails', 'update_model_yml', 'widget_filterbar',
+           'widget_image_adjustment', 'widget_media', 'widgets']

matchypatchy-0.2.1rc1/src/matchypatchy/__main__.py

@@ -0,0 +1,78 @@
+'''
+Main entry point for MatchyPatchy application
+'''
+
+import os
+import sys
+import time
+from pathlib import Path
+from PyQt6.QtWidgets import QApplication
+
+from matchypatchy.logging_config import setup_logger, get_logger
+from matchypatchy.gui import MainWindow
+
+
+def setup_cuda_path():
+    """Dynamically find and add NVIDIA CUDA/cuDNN libraries to PATH"""
+    try:
+        # Try to find nvidia packages in site-packages
+        import site
+        site_packages = site.getsitepackages()
+        
+        if isinstance(site_packages, str):
+            site_packages = [site_packages]
+        
+        for site_dir in site_packages:
+            nvidia_lib_path = Path(site_dir) / "nvidia"
+            
+            if nvidia_lib_path.exists():
+                dll_folders = [
+                    nvidia_lib_path / "cuda_runtime" / "bin",
+                    nvidia_lib_path / "cuda_nvrtc" / "bin",
+                    nvidia_lib_path / "cublas" / "bin",
+                    nvidia_lib_path / "cudnn" / "bin",
+                ]
+                
+                for folder in dll_folders:
+                    if folder.exists():
+                        os.environ['PATH'] = str(folder) + os.pathsep + os.environ['PATH']
+                
+                return True
+    except Exception as e:
+        print(f"Warning: Could not setup CUDA path: {e}")
+    
+    return False
+
+
+if __name__ == "__main__":
+    start_time = time.time()
+    
+    # Setup application-wide logging
+    root_logger = setup_logger()
+    logger = get_logger(__name__)
+    logger.info("=" * 70)
+    logger.info("MatchyPatchy starting up...")
+
+    exit_code = 1  # Default to error; overwritten on successful run
+
+    try:
+        setup_cuda_path()
+        app = QApplication(sys.argv)
+        window = MainWindow(logger)
+        logger.info("Main window initialized")
+        window.show()
+        
+        startup_time = time.time() - start_time
+        logger.info(f"Startup took {startup_time:.2f} seconds")
+        logger.info("-" * 70)
+        
+        exit_code = app.exec()
+
+    except Exception as e:
+        logger.error(f"Fatal error during startup: {e}", exc_info=True)
+        exit_code = 1
+    
+    finally:
+        logger.info("MatchyPatchy shutting down")
+        logger.info("=" * 70)
+        sys.exit(exit_code)

matchypatchy-0.2.1rc1/src/matchypatchy/assets/models.yml

@@ -0,0 +1,25 @@
+MODELS: 
+  "MegaDetector": 
+    - ["md_v1000.0.0-sorrel.onnx"]
+    - ["https://sandiegozoo.box.com/shared/static/7z9xagk8zweu6op3dgdk386jf9ayl2gt.onnx"]
+  "miewid_v3": 
+    - ["miewid_v3.onnx", "miewid_v3.onnx.data"]
+    - ["https://sandiegozoo.box.com/shared/static/ppm0odjgscvpj9lf5ax86p3fqu5vraxw.onnx",
+       "https://sandiegozoo.box.com/shared/static/xi1dijx35ta2tqtu2sa28bd09ve9li6f.data"]
+  "Jaguar Viewpoint": 
+    - ["sdzwa_jaguar_viewpoint.onnx", "sdzwa_jaguar_viewpoint.onnx.data"]
+    - ["https://sandiegozoo.box.com/shared/static/nkuku5aeicg9hugo0hogdjxg667xodlb.onnx",
+       "https://sandiegozoo.box.com/shared/static/3563emet6b6tubtgtqce8144dlq3z8y5.data"]
+
+
+DETECTOR_MODELS: 
+  - "MegaDetector"
+
+REID_MODELS: ["miewid_v3"]
+VIEWPOINT_MODELS: ["Jaguar Viewpoint"]
+
+VIEWPOINTS:
+  'Any': "Any"
+  'None': "None"
+  '0': "Right"
+  '1': "Left"