megadetector 5.0.7__py3-none-any.whl → 5.0.9__py3-none-any.whl

data_management/importers/animl_results_to_md_results.py

@@ -1,160 +1,158 @@
-########
-#
-# animl_results_to_md_results.py
-#
-# Convert a .csv file produced by the Animl package:
-#
-# https://github.com/conservationtechlab/animl-py
-#
-# ...to a MD results file suitable for import into Timelapse.
-#
-# Columns are expected to be:
-#
-# file
-# category (MD category identifies: 1==animal, 2==person, 3==vehicle)
-# detection_conf
-# bbox1,bbox2,bbox3,bbox4
-# class
-# classification_conf
-#
-########
-
-#%% Imports and constants
-
-import pandas as pd
-import json
-
-# It's a little icky to hard-code this here rather than importing from elsewhere
-# in the MD repo, but it seemed silly to take a dependency on lots of MD code
-# just for this, so, hard-coding.
-detection_category_id_to_name = {'1':'animal','2':'person','3':'vehicle'}
-
-
-#%% Main function
-
-def animl_results_to_md_results(input_file,output_file=None):
-    """
-    Converts the .csv file [input_file] to the MD-formatted .json file [output_file].
-    
-    If [output_file] is None, '.json' will be appended to the input file.
-    """
-    
-    #%%
-    
-    if output_file is None:
-        output_file = input_file + '.json'
-
-    df = pd.read_csv(input_file)
-    
-    expected_columns = ('file','category','detection_conf',
-                        'bbox1','bbox2','bbox3','bbox4','class','classification_conf')
-    
-    for s in expected_columns:
-        assert s in df.columns,\
-            'Expected column {} not found'.format(s)
-            
-    classification_category_name_to_id = {}    
-    filename_to_results = {}
-    
-    # i_row = 0; row = df.iloc[i_row]
-    for i_row,row in df.iterrows():
-            
-        # Is this the first detection we've seen for this file?            
-        if row['file'] in filename_to_results:
-            im = filename_to_results[row['file']]
-        else:
-            im = {}
-            im['detections'] = []
-            im['file'] = row['file']
-            filename_to_results[im['file']] = im
-            
-        assert isinstance(row['category'],int),'Invalid category identifier in row {}'.format(im['file'])
-        detection_category_id = str(row['category'])
-        assert detection_category_id in detection_category_id_to_name,\
-            'Unrecognized detection category ID {}'.format(detection_category_id)
-            
-        detection = {}
-        detection['category'] = detection_category_id
-        detection['conf'] = row['detection_conf']
-        bbox = [row['bbox1'],row['bbox2'],row['bbox3'],row['bbox4']]
-        detection['bbox'] = bbox
-        classification_category_name = row['class']
-        
-        # Have we seen this classification category before?
-        if classification_category_name in classification_category_name_to_id:
-            classification_category_id = \
-                classification_category_name_to_id[classification_category_name]
-        else:
-            classification_category_id = str(len(classification_category_name_to_id))
-            classification_category_name_to_id[classification_category_name] = \
-                classification_category_id
-        
-        classifications = [[classification_category_id,row['classification_conf']]]
-        detection['classifications'] = classifications
-        
-        im['detections'].append(detection)
-        
-    # ...for each row        
-        
-    info = {}
-    info['format_version'] = '1.3'
-    info['detector'] = 'Animl'
-    info['classifier'] = 'Animl'
-    
-    results = {}
-    results['info'] = info
-    results['detection_categories'] = detection_category_id_to_name
-    results['classification_categories'] = \
-        {v: k for k, v in classification_category_name_to_id.items()}
-    results['images'] = list(filename_to_results.values())
-    
-    with open(output_file,'w') as f:
-        json.dump(results,f,indent=1)
-        
-# ...zamba_results_to_md_results(...)
-        
-
-#%% Interactive driver
-
-if False:
-
-    pass
-
-    #%%
-    
-    input_file = r"G:\temp\animl-runs\animl-runs\Coati_v2\manifest.csv"
-    output_file = None
-    animl_results_to_md_results(input_file,output_file)
-    
-
-#%% Command-line driver
-
-import sys,argparse
-
-def main():
-
-    parser = argparse.ArgumentParser(
-        description='Convert an Animl-formatted .csv results file to MD-formatted .json results file')
-    
-    parser.add_argument(
-        'input_file',
-        type=str,
-        help='input .csv file')
-    
-    parser.add_argument(
-        '--output_file',
-        type=str,
-        default=None,
-        help='output .json file (defaults to input file appended with ".json")')
-    
-    if len(sys.argv[1:]) == 0:
-        parser.print_help()
-        parser.exit()
-
-    args = parser.parse_args()
-
-    animl_results_to_md_results(args.input_file,args.output_file)
-    
-if __name__ == '__main__':
-    main()
-    
+"""
+
+ animl_results_to_md_results.py
+
+ Convert a .csv file produced by the Animl package:
+
+ https://github.com/conservationtechlab/animl-py
+
+ ...to a MD results file suitable for import into Timelapse.
+
+ Columns are expected to be:
+
+ file
+ category (MD category identifies: 1==animal, 2==person, 3==vehicle)
+ detection_conf
+ bbox1,bbox2,bbox3,bbox4
+ class
+ classification_conf
+
+"""
+
+#%% Imports and constants
+
+import pandas as pd
+import json
+
+# It's a little icky to hard-code this here rather than importing from elsewhere
+# in the MD repo, but it seemed silly to take a dependency on lots of MD code
+# just for this, so, hard-coding.
+detection_category_id_to_name = {'1':'animal','2':'person','3':'vehicle'}
+
+
+#%% Main function
+
+def animl_results_to_md_results(input_file,output_file=None):
+    """
+    Converts the .csv file [input_file] to the MD-formatted .json file [output_file].
+    
+    If [output_file] is None, '.json' will be appended to the input file.
+    """
+    
+    if output_file is None:
+        output_file = input_file + '.json'
+
+    df = pd.read_csv(input_file)
+    
+    expected_columns = ('file','category','detection_conf',
+                        'bbox1','bbox2','bbox3','bbox4','class','classification_conf')
+    
+    for s in expected_columns:
+        assert s in df.columns,\
+            'Expected column {} not found'.format(s)
+            
+    classification_category_name_to_id = {}    
+    filename_to_results = {}
+    
+    # i_row = 0; row = df.iloc[i_row]
+    for i_row,row in df.iterrows():
+            
+        # Is this the first detection we've seen for this file?            
+        if row['file'] in filename_to_results:
+            im = filename_to_results[row['file']]
+        else:
+            im = {}
+            im['detections'] = []
+            im['file'] = row['file']
+            filename_to_results[im['file']] = im
+            
+        assert isinstance(row['category'],int),'Invalid category identifier in row {}'.format(im['file'])
+        detection_category_id = str(row['category'])
+        assert detection_category_id in detection_category_id_to_name,\
+            'Unrecognized detection category ID {}'.format(detection_category_id)
+            
+        detection = {}
+        detection['category'] = detection_category_id
+        detection['conf'] = row['detection_conf']
+        bbox = [row['bbox1'],row['bbox2'],row['bbox3'],row['bbox4']]
+        detection['bbox'] = bbox
+        classification_category_name = row['class']
+        
+        # Have we seen this classification category before?
+        if classification_category_name in classification_category_name_to_id:
+            classification_category_id = \
+                classification_category_name_to_id[classification_category_name]
+        else:
+            classification_category_id = str(len(classification_category_name_to_id))
+            classification_category_name_to_id[classification_category_name] = \
+                classification_category_id
+        
+        classifications = [[classification_category_id,row['classification_conf']]]
+        detection['classifications'] = classifications
+        
+        im['detections'].append(detection)
+        
+    # ...for each row        
+        
+    info = {}
+    info['format_version'] = '1.3'
+    info['detector'] = 'Animl'
+    info['classifier'] = 'Animl'
+    
+    results = {}
+    results['info'] = info
+    results['detection_categories'] = detection_category_id_to_name
+    results['classification_categories'] = \
+        {v: k for k, v in classification_category_name_to_id.items()}
+    results['images'] = list(filename_to_results.values())
+    
+    with open(output_file,'w') as f:
+        json.dump(results,f,indent=1)
+        
+# ...animl_results_to_md_results(...)
+        
+
+#%% Interactive driver
+
+if False:
+
+    pass
+
+    #%%
+    
+    input_file = r"G:\temp\animl-runs\animl-runs\Coati_v2\manifest.csv"
+    output_file = None
+    animl_results_to_md_results(input_file,output_file)
+    
+
+#%% Command-line driver
+
+import sys,argparse
+
+def main():
+
+    parser = argparse.ArgumentParser(
+        description='Convert an Animl-formatted .csv results file to MD-formatted .json results file')
+    
+    parser.add_argument(
+        'input_file',
+        type=str,
+        help='input .csv file')
+    
+    parser.add_argument(
+        '--output_file',
+        type=str,
+        default=None,
+        help='output .json file (defaults to input file appended with ".json")')
+    
+    if len(sys.argv[1:]) == 0:
+        parser.print_help()
+        parser.exit()
+
+    args = parser.parse_args()
+
+    animl_results_to_md_results(args.input_file,args.output_file)
+    
+if __name__ == '__main__':
+    main()
+    

data_management/importers/auckland_doc_test_to_json.py

@@ -1,12 +1,12 @@
-########
-#
-# auckland_doc_test_to_json.py
-#
-# Convert Auckland DOC data set to COCO camera traps format.  This was
-# for a testing data set where a .csv file was provided with class
-# information.
-#
-########
+"""
+
+ auckland_doc_test_to_json.py
+
+ Convert Auckland DOC data set to COCO camera traps format.  This was
+ for a testing data set where a .csv file was provided with class
+ information.
+
+"""
 
 #%% Constants and imports
 

data_management/importers/auckland_doc_to_json.py

@@ -1,11 +1,11 @@
-########
-#
-# auckland_doc_to_json.py
-#
-# Convert Auckland DOC data set to COCO camera traps format.  This was
-# for a training data set where class names were encoded in path names.
-#
-########
+"""
+
+ auckland_doc_to_json.py
+
+ Convert Auckland DOC data set to COCO camera traps format.  This was
+ for a training data set where class names were encoded in path names.
+
+"""
 
 #%% Constants and imports
 

data_management/importers/awc_to_json.py

@@ -1,10 +1,10 @@
-########
-#
-# awc_to_json.py
-#
-# Convert a particular .csv file from Australian Wildlife Conservancy to CCT format.
-#
-########
+"""
+
+ awc_to_json.py
+
+ Convert a particular .csv file from Australian Wildlife Conservancy to CCT format.
+
+"""
 
 #%% Constants and environment
 

data_management/importers/bellevue_to_json.py

@@ -1,18 +1,18 @@
-########
-#
-# bellevue_to_json.py
-#
-# "Bellevue Camera Traps" is the rather unremarkable camera trap data set
-# used by one of the repo's maintainers for testing.  It's organized as:
-#
-# approximate_date/[loose_camera_specifier/]/species    
-#
-# E.g.:
-#    
-# "2018.03.30\coyote\DSCF0091.JPG"
-# "2018.07.18\oldcam\empty\DSCF0001.JPG"
-#
-########
+"""
+
+ bellevue_to_json.py
+
+ "Bellevue Camera Traps" is the rather unremarkable camera trap data set
+ used by one of the repo's maintainers for testing.  It's organized as:
+
+ approximate_date/[loose_camera_specifier/]/species    
+
+ E.g.:
+    
+ "2018.03.30\coyote\DSCF0091.JPG"
+ "2018.07.18\oldcam\empty\DSCF0001.JPG"
+
+"""
 
 #%% Constants and imports
 

data_management/importers/cacophony-thermal-importer.py

@@ -1,16 +1,16 @@
-########
-#
-# cacophony-thermal-importer.py
-#
-# Create data and metadata for LILA from the Cacophony thermal dataset.  Takes a folder
-# of HDF files, and produces .json metadata, along with compressed/normalized videos for
-# each HDF file.
-#
-# Source format notes for this dataset:
-#    
-# https://docs.google.com/document/d/12sw5JtwdMf9MiXuNCBcvhvZ04Jwa1TH2Lf6LnJmF8Bk/edit
-#
-########
+"""
+
+ cacophony-thermal-importer.py
+
+ Create data and metadata for LILA from the Cacophony thermal dataset.  Takes a folder
+ of HDF files, and produces .json metadata, along with compressed/normalized videos for
+ each HDF file.
+
+ Source format notes for this dataset:
+    
+ https://docs.google.com/document/d/12sw5JtwdMf9MiXuNCBcvhvZ04Jwa1TH2Lf6LnJmF8Bk/edit
+
+"""
 
 #%% Imports and constants
 

data_management/importers/carrizo_shrubfree_2018.py

@@ -1,11 +1,11 @@
-########
-#
-# carrizo_shrubfree_2018.py
-#
-# Convert the .csv file provided for the Carrizo Mojave data set to a 
-# COCO-camera-traps .json file
-#
-########
+"""
+
+ carrizo_shrubfree_2018.py
+
+ Convert the .csv file provided for the Carrizo Mojave data set to a 
+ COCO-camera-traps .json file
+
+"""
 
 #%% Constants and environment
 

data_management/importers/carrizo_trail_cam_2017.py

@@ -1,11 +1,11 @@
-########
-#
-# carrizo_trail_cam_2017.py
-#
-# Convert the .csv files provided for the "Trail Cam Carrizo" 2017 data set to 
-# a COCO-camera-traps .json file.
-#
-########
+"""
+
+ carrizo_trail_cam_2017.py
+
+ Convert the .csv files provided for the "Trail Cam Carrizo" 2017 data set to 
+ a COCO-camera-traps .json file.
+
+"""
 
 #%% Constants and environment
 

data_management/importers/cct_field_adjustments.py

@@ -1,12 +1,12 @@
-########
-#
-# cct_field_adjustments.py
-#
-# CCT metadata was posted with int locations instead of strings.
-#
-# This script fixes those issues and rev's the version number.
-#
-########
+"""
+
+ cct_field_adjustments.py
+
+ CCT metadata was posted with int locations instead of strings.
+
+ This script fixes those issues and rev's the version number.
+
+"""
 
 #%% Constants and environment
 

data_management/importers/channel_islands_to_cct.py

@@ -1,13 +1,13 @@
-########
-#
-# channel_islands_to_cct.py
-#
-# Convert the Channel Islands data set to a COCO-camera-traps .json file
-#
-# Uses the command-line tool ExifTool (exiftool.org) to pull EXIF tags from images,
-# because every Python package we tried failed to pull the "Maker Notes" field properly.
-#
-########
+"""
+
+ channel_islands_to_cct.py
+
+ Convert the Channel Islands data set to a COCO-camera-traps .json file
+
+ Uses the command-line tool ExifTool (exiftool.org) to pull EXIF tags from images,
+ because every Python package we tried failed to pull the "Maker Notes" field properly.
+
+"""
 
 #%% Imports, constants, paths
 

data_management/importers/eMammal/copy_and_unzip_emammal.py

@@ -155,6 +155,7 @@ def download_from_container(dest_folder: str,
 #%% Command-line driver
         
 if __name__ == '__main__':
+    
     if origin == 'cloud':
         container = 'wpz'
         desired_blob_prefix = 'emammal/0Robert Long/'

data_management/importers/ena24_to_json.py

@@ -1,10 +1,10 @@
-########
-#
-# ena24_to_json_2017.py
-#
-# Convert the ENA24 data set to a COCO-camera-traps .json file
-#
-########
+"""
+
+ ena24_to_json_2017.py
+
+ Convert the ENA24 data set to a COCO-camera-traps .json file
+
+"""
 
 #%% Constants and environment
 

data_management/importers/filenames_to_json.py

@@ -1,11 +1,11 @@
-########
-#
-# filenames_to_json.py
-#
-# Take a directory of images in which species labels are encoded by folder
-# names, and produces a COCO-style .json file 
-# 
-########
+"""
+
+ filenames_to_json.py
+
+ Take a directory of images in which species labels are encoded by folder
+ names, and produces a COCO-style .json file 
+ 
+"""
 
 #%% Constants and imports
 

data_management/importers/helena_to_cct.py

@@ -1,10 +1,10 @@
-########
-#
-# helena_to_cct.py
-#
-# Convert the Helena Detections data set to a COCO-camera-traps .json file
-#
-########
+"""
+
+ helena_to_cct.py
+
+ Convert the Helena Detections data set to a COCO-camera-traps .json file
+
+"""
 
 #%% Constants and environment
 

data_management/importers/idaho-camera-traps.py

@@ -1,10 +1,10 @@
-########
-#
-# idaho-camera-traps.py
-#
-# Prepare the Idaho Camera Traps dataset for release on LILA.
-#
-########
+"""
+
+ idaho-camera-traps.py
+
+ Prepare the Idaho Camera Traps dataset for release on LILA.
+
+"""
 
 #%% Imports and constants
 

data_management/importers/idfg_iwildcam_lila_prep.py

@@ -1,13 +1,13 @@
-########
-#
-# idfg_iwildcam_lila_prep.py
-#
-# Adding class labels (from the private test .csv) to the iWildCam 2019 IDFG 
-# test set, in preparation for release on LILA.
-#
-# This version works with the public iWildCam release images.
-#
-########
+"""
+
+ idfg_iwildcam_lila_prep.py
+
+ Adding class labels (from the private test .csv) to the iWildCam 2019 IDFG 
+ test set, in preparation for release on LILA.
+
+ This version works with the public iWildCam release images.
+
+"""
 
 #%% ############ Take one, from iWildCam .json files ############