1

ILG2021 · Nov 13, 2022 · d4eb71f · d4eb71f
1 parent 0306981
commit d4eb71f
Show file tree

Hide file tree

Showing 6 changed files with 2 additions and 44 deletions.
diff --git a/Dockerfile b/Dockerfile
diff --git a/additional_requirements.txt b/additional_requirements.txt
diff --git a/app.yaml b/app.yaml
diff --git a/pyinstaller.bat b/pyinstaller.bat
@@ -0,0 +1 @@
+pyinstaller --noconfirm --onedir --console --icon "./yolo.png" --add-data "./venv/Lib/site-packages/cv2;cv2/" --add-data "./venv/Lib/site-packages/yaml;yaml/" --hidden-import "PIL.ExifTags" --add-data "./venv/Lib/site-packages/seaborn;seaborn/"  "./yolov5.py"
diff --git a/yolo.png b/yolo.png
diff --git a/api.py → yolov5.py b/api.py → yolov5.py
@@ -29,7 +29,7 @@ def analyze():
         for instance in instances:
             encoded_data = instance.split(',')[1]
             im = Image.open(io.BytesIO(base64.b64decode(encoded_data)))
-            ai_results = models['yolov5s'](im, size=120)  # reduce size=320 for faster inference
+            ai_results = models['yolov5s'](im, size=320)  # reduce size=320 for faster inference
             input_dict = json.loads(ai_results.pandas().xyxy[0].to_json(orient="records"))
             output_dict = [x for x in input_dict if x['name'] == 'person']
             results.append(len(output_dict))
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		pyinstaller --noconfirm --onedir --console --icon "./yolo.png" --add-data "./venv/Lib/site-packages/cv2;cv2/" --add-data "./venv/Lib/site-packages/yaml;yaml/" --hidden-import "PIL.ExifTags" --add-data "./venv/Lib/site-packages/seaborn;seaborn/" "./yolov5.py"