Fix maxBuffer length limit (#40)

engine/cli_wrapper/cli.py

@@ -6,11 +6,11 @@
 from utils import save_as_csv
 
 
+
 @click.group()
 def mega():
     pass
 
-
 # TODO: support for results field
 
 
@@ -25,7 +25,7 @@ def mega():
 @click.option('--recursive/--not-recursive', default=False, help='Whether to search for images in folders within the base folder provided')
 @click.option('-n', '--n-cores', default=0, help='Number of CPU cores to utilise. Will be ignored if a valid GPU is available')
 @click.option('-cp', '--checkpoint-path', default=None, type=str, help='Path to JSON checkpoint file')
-@click.option('-cf', '--checkpoint-frequency', default=-1, type=str, help='How often to write to checkpoint file, i.e.: every N images')
+@click.option('-cf', '--checkpoint-frequency', default=-1, type=int, help='How often to write to checkpoint file, i.e.: every N images')
 @click.option('--show/--no-show', default=False, help='Whether to output the results in the console')
 @click.option('--bbox/--no-bbox', default=True, help='Whether save images with bounding boxes.')
 @click.option('--verbose/--quiet', default=False, help='Whether to output or supress Tensorflow message')
@@ -108,5 +108,10 @@ def move(results_path, auto_sort):
             continue
 
 
+# Use this when testing out this script from the command line, i.e.: python cli.py detect ....
+# if __name__ == '__main__':
+#     mega()
+
+# Use this when invoking from .exe
 if getattr(sys, 'frozen', False):
     mega(sys.argv[1:])

engine/engine_env.yml

@@ -0,0 +1,21 @@
+name: engine_env
+
+channels:
+  - conda-forge
+  - defaults
+
+dependencies:
+  - cudatoolkit=10.1
+  - cudnn=7.6
+  - python=3.7
+  - tensorflow-gpu=2.3
+  - numpy
+  - Pillow
+  - requests
+  - setuptools
+  - click
+  - pyinstaller
+  - matplotlib
+  - pip
+  - pip:
+    - exif

engine/requirements.txt

@@ -1,8 +1,10 @@
-numpy==1.18.5
-Pillow==8.0.1
-requests==2.25.0
-tensorflow==2.3.1
-tensorflow-gpu~=2.3.1
-setuptools~=50.3.2
-click~=7.1.2
-pyinstaller
+numpy
+Pillow
+requests
+tensorflow
+tensorflow-gpu
+setuptools
+click
+pyinstaller
+# exif
+matplotlib