Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-19 23:08:08 +02:00
Code Issues Projects Releases Wiki Activity

Cleanup onnx detector (#20128)

Browse Source 
* Cleanup onnx detector

* Fix

* Fix classification cropping

* Deprioritize openvino

* Send model type

* Use model type to decide if model can use full optimization

* Clenanup

* Cleanup
This commit is contained in:
Nicolas Mowen
2025-09-18 15:12:09 -06:00
committed by GitHub
parent c05e260ae9
commit 68f806bb61
10 changed files with 82 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
frigate/data_processing/real_time/custom_classification.py
View File

@@ -133,9 +133,9 @@ class CustomStateClassificationProcessor(RealTimeProcessorApi):
x:x2,
]
if input.shape != (224, 224):
if frame.shape != (224, 224):
try:
input = cv2.resize(input, (224, 224))
frame = cv2.resize(frame, (224, 224))
except Exception:
logger.warning("Failed to resize image for state classification")
return
@@ -258,11 +258,11 @@ class CustomObjectClassificationProcessor(RealTimeProcessorApi):
x:x2,
]
if input.shape != (224, 224):
if crop.shape != (224, 224):
try:
input = cv2.resize(input, (224, 224))
crop = cv2.resize(crop, (224, 224))
except Exception:
logger.warning("Failed to resize image for object classification")
logger.warning("Failed to resize image for state classification")
return
input = np.expand_dims(crop, axis=0)