megadetector 5.0.10__tar.gz → 5.0.12__tar.gz

{megadetector-5.0.10/megadetector.egg-info → megadetector-5.0.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: megadetector
-Version: 5.0.10
+Version: 5.0.12
 Summary: MegaDetector is an AI model that helps conservation folks spend less time doing boring things with camera trap images.
 Author-email: Your friendly neighborhood MegaDetector team <cameratraps@lila.science>
 Maintainer-email: Your friendly neighborhood MegaDetector team <cameratraps@lila.science>
@@ -28,7 +28,7 @@ Project-URL: Homepage, https://github.com/agentmorris/MegaDetector
 Project-URL: Documentation, https://megadetector.readthedocs.io
 Project-URL: Bug Reports, https://github.com/agentmorris/MegaDetector/issues
 Project-URL: Source, https://github.com/agentmorris/MegaDetector
-Keywords: camera traps,conservation,wildlife,ai
+Keywords: camera traps,conservation,wildlife,ai,megadetector
 Classifier: Development Status :: 3 - Alpha
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
@@ -52,9 +52,9 @@ Requires-Dist: ultralytics-yolov5==0.1.1
 
 # MegaDetector
 
-This package is a pip-installable version of the support/inference code for [MegaDetector](https://github.com/agentmorris/MegaDetector), an object detection model that helps conservation biologists spend less time doing boring things with camera trap images.  Complete documentation for this Python package is available at [megadetector.readthedocs.io](https://megadetector.readthedocs.io).
+This package is a pip-installable version of the support/inference code for [MegaDetector](https://github.com/agentmorris/MegaDetector/?tab=readme-ov-file#megadetector), an object detection model that helps conservation biologists spend less time doing boring things with camera trap images.  Complete documentation for this Python package is available at [megadetector.readthedocs.io](https://megadetector.readthedocs.io).
 
-If you aren't looking for the Python package specificaly, and you just want to learn more about what MegaDetector is all about, head over to the [MegaDetector repo](https://github.com/agentmorris/MegaDetector).
+If you aren't looking for the Python package specificaly, and you just want to learn more about what MegaDetector is all about, head over to the [MegaDetector repo](https://github.com/agentmorris/MegaDetector/?tab=readme-ov-file#megadetector).
 
 
 ## Reasons you probably aren't looking for this package
@@ -69,7 +69,7 @@ If you are a computer-vision-y person looking to run or fine-tune MegaDetector p
 
 ## Reasons you might want to use this package
 
-If you want to programmatically interact with the postprocessing tools from the MegaDetector repo, or programmatically run MegaDetector in a way that produces [Timelapse](https://saul.cpsc.ucalgary.ca/timelapse)-friendly output (i.e., output in the standard [MegaDetector output format](https://github.com/agentmorris/MegaDetector/tree/main/api/batch_processing#megadetector-batch-output-format)), this package might be for you.
+If you want to programmatically interact with the postprocessing tools from the MegaDetector repo, or programmatically run MegaDetector in a way that produces [Timelapse](https://saul.cpsc.ucalgary.ca/timelapse)-friendly output (i.e., output in the standard [MegaDetector output format](https://github.com/agentmorris/MegaDetector/tree/main/megadetector/api/batch_processing#megadetector-batch-output-format)), this package might be for you.
 
 Although even if that describes you, you <i>still</i> might be better off cloning the MegaDetector repo.  Pip-installability requires that some dependencies be newer than what was available at the time MDv5 was trained, so results are <i>very slightly</i> different than results produced in the "official" environment.  These differences <i>probably</i> don't matter much, but they have not been formally characterized.
 
@@ -83,7 +83,7 @@ MegaDetector model weights aren't downloaded at pip-install time, but they will
 
 ## Package reference
 
-See <megadetector.readthedocs.io>.
+See [megadetector.readthedocs.io](https://megadetector.readthedocs.io).
 
 
 ## Examples of things you can do with this package
@@ -91,9 +91,9 @@ See <megadetector.readthedocs.io>.
 ### Run MegaDetector on one image and count the number of detections
 
 ```
-from md_utils import url_utils
-from md_visualization import visualization_utils as vis_utils
-from detection import run_detector
+from megadetector.utils import url_utils
+from megadetector.visualization import visualization_utils as vis_utils
+from megadetector.detection import run_detector
 
 # This is the image at the bottom of this page, it has one animal in it
 image_url = 'https://github.com/agentmorris/MegaDetector/raw/main/images/orinoquia-thumb-web.jpg'
@@ -113,8 +113,9 @@ print('Found {} detections above threshold'.format(len(detections_above_threshol
 ### Run MegaDetector on a folder of images
 
 ```
-from detection.run_detector_batch import load_and_run_detector_batch,write_results_to_file
-from md_utils import path_utils
+from megadetector.detection.run_detector_batch import \
+    load_and_run_detector_batch,write_results_to_file
+from megadetector.utils import path_utils
 import os
 
 # Pick a folder to run MD on recursively, and an output file

{megadetector-5.0.10 → megadetector-5.0.12}/README-package.md

@@ -1,8 +1,8 @@
 # MegaDetector
 
-This package is a pip-installable version of the support/inference code for [MegaDetector](https://github.com/agentmorris/MegaDetector), an object detection model that helps conservation biologists spend less time doing boring things with camera trap images.  Complete documentation for this Python package is available at [megadetector.readthedocs.io](https://megadetector.readthedocs.io).
+This package is a pip-installable version of the support/inference code for [MegaDetector](https://github.com/agentmorris/MegaDetector/?tab=readme-ov-file#megadetector), an object detection model that helps conservation biologists spend less time doing boring things with camera trap images.  Complete documentation for this Python package is available at [megadetector.readthedocs.io](https://megadetector.readthedocs.io).
 
-If you aren't looking for the Python package specificaly, and you just want to learn more about what MegaDetector is all about, head over to the [MegaDetector repo](https://github.com/agentmorris/MegaDetector).
+If you aren't looking for the Python package specificaly, and you just want to learn more about what MegaDetector is all about, head over to the [MegaDetector repo](https://github.com/agentmorris/MegaDetector/?tab=readme-ov-file#megadetector).
 
 
 ## Reasons you probably aren't looking for this package
@@ -17,7 +17,7 @@ If you are a computer-vision-y person looking to run or fine-tune MegaDetector p
 
 ## Reasons you might want to use this package
 
-If you want to programmatically interact with the postprocessing tools from the MegaDetector repo, or programmatically run MegaDetector in a way that produces [Timelapse](https://saul.cpsc.ucalgary.ca/timelapse)-friendly output (i.e., output in the standard [MegaDetector output format](https://github.com/agentmorris/MegaDetector/tree/main/api/batch_processing#megadetector-batch-output-format)), this package might be for you.
+If you want to programmatically interact with the postprocessing tools from the MegaDetector repo, or programmatically run MegaDetector in a way that produces [Timelapse](https://saul.cpsc.ucalgary.ca/timelapse)-friendly output (i.e., output in the standard [MegaDetector output format](https://github.com/agentmorris/MegaDetector/tree/main/megadetector/api/batch_processing#megadetector-batch-output-format)), this package might be for you.
 
 Although even if that describes you, you <i>still</i> might be better off cloning the MegaDetector repo.  Pip-installability requires that some dependencies be newer than what was available at the time MDv5 was trained, so results are <i>very slightly</i> different than results produced in the "official" environment.  These differences <i>probably</i> don't matter much, but they have not been formally characterized.
 
@@ -31,7 +31,7 @@ MegaDetector model weights aren't downloaded at pip-install time, but they will
 
 ## Package reference
 
-See <megadetector.readthedocs.io>.
+See [megadetector.readthedocs.io](https://megadetector.readthedocs.io).
 
 
 ## Examples of things you can do with this package
@@ -39,9 +39,9 @@ See <megadetector.readthedocs.io>.
 ### Run MegaDetector on one image and count the number of detections
 
 ```
-from md_utils import url_utils
-from md_visualization import visualization_utils as vis_utils
-from detection import run_detector
+from megadetector.utils import url_utils
+from megadetector.visualization import visualization_utils as vis_utils
+from megadetector.detection import run_detector
 
 # This is the image at the bottom of this page, it has one animal in it
 image_url = 'https://github.com/agentmorris/MegaDetector/raw/main/images/orinoquia-thumb-web.jpg'
@@ -61,8 +61,9 @@ print('Found {} detections above threshold'.format(len(detections_above_threshol
 ### Run MegaDetector on a folder of images
 
 ```
-from detection.run_detector_batch import load_and_run_detector_batch,write_results_to_file
-from md_utils import path_utils
+from megadetector.detection.run_detector_batch import \
+    load_and_run_detector_batch,write_results_to_file
+from megadetector.utils import path_utils
 import os
 
 # Pick a folder to run MD on recursively, and an output file

{megadetector-5.0.10 → megadetector-5.0.12}/README.md

@@ -147,25 +147,40 @@ The core functionality provided is:
 - Training and running [MegaDetector](megadetector.md).
 - Training and running some [species classifiers](classification) that are used in conjunction with MegaDetector.
 - Tools to [convert](data_management) frequently-used camera trap metadata formats into a common format.
-- A [batch processing API](https://github.com/agentmorris/MegaDetector/tree/main/api/batch_processing) that runs MegaDetector on large image collections, to accelerate population surveys.
-- A [real-time API](https://github.com/agentmorris/MegaDetector/tree/main/api/synchronous) that runs MegaDetector (and some species classifiers) synchronously, primarily to support biosecurity applications.
+- A [batch processing API](https://github.com/agentmorris/MegaDetector/tree/main/megadetector/api/batch_processing) that runs MegaDetector on large image collections, to accelerate population surveys.
+- A [real-time API](https://github.com/agentmorris/MegaDetector/tree/main/megadetector/api/synchronous) that runs MegaDetector (and some species classifiers) synchronously, primarily to support biosecurity applications.
 
 This repo does not host the data used to train MegaDetector, but we work with our collaborators to make data and annotations available whenever possible on [lila.science](http://lila.science).  See the [MegaDetector training data](megadetector.md#can-you-share-the-training-data) section to learn more about the data used to train MegaDetector.
 
 MegaDetector was initially developed by the [Microsoft AI for Earth program](https://www.microsoft.com/en-us/ai/ai-for-earth); this repo was forked from the microsoft/cameratraps repo and is maintained by the original MegaDetector developers (who are no longer at Microsoft, but are absolutely fantastically eternally grateful to Microsoft for the investment and commitment that made MegaDetector happen).
 
 
-### api
+### megadetector/detection
 
-Code for hosting our models as an API, either for synchronous operation (i.e., for real-time inference) or as a batch process (for large biodiversity surveys).  Common operations one might do after running MegaDetector – e.g. [generating preview pages to summarize your results](https://github.com/agentmorris/MegaDetector/blob/main/api/batch_processing/postprocessing/postprocess_batch_results.py), [separating images into different folders based on AI results](https://github.com/agentmorris/MegaDetector/blob/main/api/batch_processing/postprocessing/separate_detections_into_folders.py), or [converting results to a different format](https://github.com/agentmorris/MegaDetector/blob/main/api/batch_processing/postprocessing/convert_output_format.py) – also live in this folder, within the [api/batch_processing/postprocessing](https://github.com/agentmorris/MegaDetector/tree/main/api/batch_processing/postprocessing) folder.
+Code for running models, especially MegaDetector.
 
 
-### archive
+### megadetector/postprocessing
+
+Code for common operations one might do after running MegaDetector, e.g. [generating preview pages to summarize your results](https://github.com/agentmorris/MegaDetector/blob/main/megadetector/postprocessing/postprocess_batch_results.py), [separating images into different folders based on AI results](https://github.com/agentmorris/MegaDetector/blob/main/megadetector/postprocessing/separate_detections_into_folders.py), or [converting results to a different format](https://github.com/agentmorris/MegaDetector/blob/main/megadetector/postprocessing/convert_output_format.py).
+
+
+### megadetector/utils
+
+Small utility functions for string manipulation, filename manipulation, downloading files from URLs, etc.  Mostly adapted from the [ai4eutils](https://github.com/microsoft/ai4eutils) repo.
 
-Old code that we didn't <i>quite</i> want to delete, but is basically obsolete.
 
+### megadetector/visualization
 
-### classification
+Tools for visualizing images with ground truth and/or predicted annotations.
+
+
+### megadetector/api
+
+Code for hosting our models as an API, either for synchronous operation (i.e., for real-time inference) or as a batch process (for large biodiversity surveys).
+
+
+### megadetector/classification
 
 Experimental code for training species classifiers on new data sets, generally trained on MegaDetector crops.  Currently the main pipeline described in this folder relies on a large database of labeled images that is not publicly available; therefore, this folder is not yet set up to facilitate training of your own classifiers.  However, it is useful for <i>users</i> of the classifiers that we train, and contains some useful starting points if you are going to take a "DIY" approach to training classifiers on cropped images.  
 
@@ -174,18 +189,18 @@ All that said, here's another "teaser image" of what you get at the end of train
 <img src="images/warthog_classifications.jpg" width="700"><br/>Image credit University of Minnesota, from the Snapshot Safari program.
 
 
-### data_management
+### megadetector/taxonomy_mapping
 
-Code for:
+Code to facilitate mapping data-set-specific category names (e.g. "lion", which means very different things in Idaho vs. South Africa) to a standard taxonomy.
 
-* Converting frequently-used metadata formats to [COCO Camera Traps](https://github.com/agentmorris/MegaDetector/blob/main/data_management/README.md#coco-cameratraps-format) format
-* Converting the output of AI models (especially [YOLOv5](https://github.com/agentmorris/MegaDetector/blob/main/api/batch_processing/postprocessing/convert_output_format.py)) to the format used for AI results throughout this repo
-* Creating, visualizing, and  editing COCO Camera Traps .json databases
 
+### megadetector/data_management
 
-### detection
+Code for:
 
-Code for running models, especially MegaDetector.
+* Converting frequently-used metadata formats to [COCO Camera Traps](https://github.com/agentmorris/MegaDetector/blob/main/megadetector/data_management/README.md#coco-cameratraps-format) format
+* Converting the output of AI models (especially YOLOv5) to the format used for AI results throughout this repo
+* Creating, visualizing, and  editing COCO Camera Traps .json databases
 
 
 ### envs
@@ -198,14 +213,9 @@ Environment files... specifically .yml files for mamba/conda environments (these
 Media used in documentation.
 
 
-### md_utils
-
-Small utility functions for string manipulation, filename manipulation, downloading files from URLs, etc.  Mostly adapted from the [ai4eutils](https://github.com/microsoft/ai4eutils) repo.
-
-
-### md_visualization
+### archive
 
-Tools for visualizing images with ground truth and/or predicted annotations.
+Old code that we didn't <i>quite</i> want to delete, but is basically obsolete.
 
 
 ### sandbox
@@ -213,11 +223,6 @@ Tools for visualizing images with ground truth and/or predicted annotations.
 Random things that don't fit in any other directory, but aren't quite deprecated.  Mostly postprocessing scripts that were built for a single use case but could potentially be useful in the future.
 
 
-### taxonomy_mapping
-
-Code to facilitate mapping data-set-specific category names (e.g. "lion", which means very different things in Idaho vs. South Africa) to a standard taxonomy.
-
-
 ### test_images
 
 A handful of images from [LILA](https://lila.science) that facilitate testing and debugging.

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/api/batch_processing/api_core/batch_service/score.py

@@ -19,7 +19,7 @@ print('score.py, tf.test.is_gpu_available:', tf.test.is_gpu_available())
 PRINT_EVERY = 500
 
 
-#%% Helper functions *copied* from ct_utils.py and md_visualization/visualization_utils.py
+#%% Helper functions *copied* from ct_utils.py and visualization/visualization_utils.py
 
 IMAGE_ROTATIONS = {
     3: 180,
@@ -237,7 +237,7 @@ class TFDetector:
             image_id: a path to identify the image; will be in the "file" field of the output object
             detection_threshold: confidence above which to include the detection proposal
         Returns:
-        A dict with the following fields, see the 'images' key in https://github.com/agentmorris/MegaDetector/tree/master/api/batch_processing#batch-processing-api-output-format
+        A dict with the following fields, see the 'images' key in https://github.com/agentmorris/MegaDetector/tree/main/megadetector/api/batch_processing#batch-processing-api-output-format
             - 'file' (always present)
             - 'max_detection_conf'
             - 'detections', which is a list of detection objects containing keys 'category', 'conf' and 'bbox'

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/api/synchronous/api_core/animal_detection_api/api_backend.py

@@ -18,7 +18,7 @@ from io import BytesIO
 
 from detection.run_detector import load_detector, convert_to_tf_coords
 import config
-import md_visualization.visualization_utils as vis_utils 
+import visualization.visualization_utils as vis_utils 
 
 #%% Initialization
 

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/api/synchronous/api_core/animal_detection_api/api_frontend.py

@@ -21,7 +21,7 @@ from io import BytesIO
 from flask import Flask, Response, jsonify, make_response, request
 from requests_toolbelt.multipart.encoder import MultipartEncoder
 
-import md_visualization.visualization_utils as vis_utils
+import visualization.visualization_utils as vis_utils
 import config
 
 

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/classification/analyze_failed_images.py

@@ -18,9 +18,9 @@ from PIL import Image, ImageFile
 import requests
 from tqdm import tqdm
 
-from data_management.megadb.megadb_utils import MegadbUtils
-from md_utils import path_utils
-from md_utils import sas_blob_utils
+from megadetector.data_management.megadb.megadb_utils import MegadbUtils
+from megadetector.utils import path_utils
+from megadetector.utils import sas_blob_utils
 
 
 #%% Example usage

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/classification/cache_batchapi_outputs.py

@@ -46,7 +46,7 @@ Detections JSON format:
 
 Batch Detection API Output Format:
 
-github.com/agentmorris/MegaDetector/tree/master/api/batch_processing#api-outputs
+github.com/agentmorris/MegaDetector/tree/main/megadetector/api/batch_processing#api-outputs
 
 """
 

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/classification/create_classification_dataset.py

@@ -74,7 +74,7 @@ import numpy as np
 import pandas as pd
 from tqdm import tqdm
 
-from classification import detect_and_crop
+from megadetector.classification import detect_and_crop
 
 
 #%% Example usage

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/classification/crop_detections.py

@@ -9,7 +9,7 @@ This script takes as input a detections JSON file, usually the output of
 detection/run_tf_detector_batch.py or the output of the Batch API in the
 "Batch processing API output format".
 
-See https://github.com/agentmorris/MegaDetector/tree/master/api/batch_processing.
+See https://github.com/agentmorris/MegaDetector/tree/main/megadetector/api/batch_processing.
 
 The script can crop images that are either available locally or that need to be
 downloaded from an Azure Blob Storage container.

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/classification/detect_and_crop.py

@@ -113,11 +113,11 @@ from tqdm import tqdm
 
 from api.batch_processing.data_preparation.prepare_api_submission import (
     BatchAPIResponseError, Task, TaskStatus, divide_list_into_tasks)
-from classification.cache_batchapi_outputs import cache_detections
-from classification.crop_detections import load_and_crop
-from data_management.megadb import megadb_utils
-from md_utils import path_utils
-from md_utils import sas_blob_utils
+from megadetector.classification.cache_batchapi_outputs import cache_detections
+from megadetector.classification.crop_detections import load_and_crop
+from megadetector.data_management.megadb import megadb_utils
+from megadetector.utils import path_utils
+from megadetector.utils import sas_blob_utils
 
 
 #%% Example usage

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/classification/evaluate_model.py

@@ -51,7 +51,7 @@ import torch
 import torchvision
 import tqdm
 
-from classification import efficientnet, train_classifier
+from megadetector.classification import efficientnet, train_classifier
 
 
 #%% Example usage

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/classification/json_to_azcopy_list.py

@@ -16,8 +16,8 @@ import os
 
 from tqdm import tqdm
 
-from data_management.megadb import megadb_utils
-from md_utils import sas_blob_utils
+from megadetector.data_management.megadb import megadb_utils
+from megadetector.utils import sas_blob_utils
 
 images_dir = ''
 queried_images_json_path = 'run_idfg2/queried_images.json'

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/classification/json_validator.py

@@ -65,26 +65,30 @@ exist in Azure Blob Storage. In total, we output the following files:
 
 """
 
+#%% Imports
+
 from __future__ import annotations
 
 import argparse
-from collections import defaultdict
-from collections.abc import Container, Iterable, Mapping, MutableMapping
-from concurrent import futures
-from datetime import datetime
 import json
 import os
 import pprint
 import random
-from typing import Any
 
 import pandas as pd
-from md_utils import path_utils
-from md_utils import sas_blob_utils
+
 from tqdm import tqdm
+from collections import defaultdict
+from collections.abc import Container, Iterable, Mapping, MutableMapping
+from concurrent import futures
+from datetime import datetime
+from typing import Any
+
+from megadetector.utils import path_utils
+from megadetector.utils import sas_blob_utils
 
-from data_management.megadb import megadb_utils
-from taxonomy_mapping.taxonomy_graph import (
+from megadetector.data_management.megadb import megadb_utils
+from megadetector.taxonomy_mapping.taxonomy_graph import (
     build_taxonomy_graph, dag_to_tree, TaxonNode)
 
 

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/classification/map_classification_categories.py

@@ -52,7 +52,7 @@ import networkx as nx
 import pandas as pd
 from tqdm import tqdm
 
-from taxonomy_mapping.taxonomy_graph import (
+from megadetector.taxonomy_mapping.taxonomy_graph import (
     build_taxonomy_graph, dag_to_tree, TaxonNode)
 
 

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/classification/merge_classification_detection_output.py

@@ -70,7 +70,7 @@ from typing import Any
 import pandas as pd
 from tqdm import tqdm
 
-from md_utils.ct_utils import truncate_float
+from megadetector.utils.ct_utils import truncate_float
 
 
 #%% Example usage

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/classification/run_classifier.py

@@ -25,6 +25,7 @@ from __future__ import annotations
 import argparse
 import json
 import os
+
 from tqdm import tqdm
 from typing import Any
 from collections.abc import Callable, Sequence
@@ -36,7 +37,7 @@ import torch.utils
 import torchvision as tv
 from torchvision.datasets.folder import default_loader
 
-from classification import train_classifier
+from megadetector.classification import train_classifier
 
 
 #%% Example usage

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/classification/train_classifier.py

@@ -16,28 +16,30 @@ During training, start tensorboard from within the classification/ directory:
 
 from __future__ import annotations
 
+import json
+import os
 import argparse
+
 from collections import defaultdict
 from collections.abc import Callable, Mapping, MutableMapping, Sequence
 from datetime import datetime
-import json
-import os
 from typing import Any
 
 import numpy as np
 import PIL.Image
 import sklearn.metrics
+import tqdm
+
 import torch
 from torch.utils import tensorboard
 import torchvision as tv
 from torchvision.datasets.folder import default_loader
-import tqdm
 
-from classification import efficientnet, evaluate_model
-from classification.train_utils import (
+from megadetector.classification import efficientnet, evaluate_model
+from megadetector.classification.train_utils import (
     HeapItem, recall_from_confusion_matrix, add_to_heap, fig_to_img,
     imgs_with_confidences, load_dataset_csv, prefix_all_keys)
-from md_visualization import plot_utils
+from megadetector.visualization import plot_utils
 
 
 #%% Example usage

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/classification/train_classifier_tf.py

@@ -16,25 +16,26 @@ During training, start tensorboard from within the classification/ directory:
 
 from __future__ import annotations
 
-import argparse
-from collections import defaultdict
-from collections.abc import Callable, Mapping, MutableMapping, Sequence
-from datetime import datetime
-import json
 import os
-from typing import Any, Optional
+import json
 import uuid
+import argparse
 
+import tqdm
 import numpy as np
 import sklearn.metrics
 import tensorflow as tf
+
+from collections import defaultdict
+from collections.abc import Callable, Mapping, MutableMapping, Sequence
+from datetime import datetime
+from typing import Any, Optional
 from tensorboard.plugins.hparams import api as hp
-import tqdm
 
-from classification.train_utils import (
+from megadetector.classification.train_utils import (
     HeapItem, recall_from_confusion_matrix, add_to_heap, fig_to_img,
     imgs_with_confidences, load_dataset_csv, prefix_all_keys)
-from md_visualization import plot_utils
+from megadetector.visualization import plot_utils
 
 AUTOTUNE = tf.data.experimental.AUTOTUNE
 

{megadetector-5.0.10 → megadetector-5.0.12/megadetector}/classification/train_utils.py

@@ -13,12 +13,10 @@ especially be agnostic to PyTorch vs. TensorFlow.
 
 from __future__ import annotations
 
-from collections.abc import Mapping, Sequence
 import dataclasses
 import heapq
 import io
 import json
-from typing import Any, Optional
 
 import matplotlib.figure
 import numpy as np
@@ -26,6 +24,9 @@ import pandas as pd
 import PIL.Image
 import scipy.interpolate
 
+from collections.abc import Mapping, Sequence
+from typing import Any, Optional
+
 
 #%% Classes