Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-12-23 19:11:14 +01:00
Code Issues Projects Releases Wiki Activity

fix scaling

Browse Source
This commit is contained in:
Blake Blackshear 2021-08-09 07:46:28 -05:00
parent 8fd12f001b
commit 1823bd0305
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
frigate/video.py
View File

@ -398,16 +398,17 @@ def detect(
object_detector, frame, model_shape, region, objects_to_track, object_filters object_detector, frame, model_shape, region, objects_to_track, object_filters
): ):
tensor_input = create_tensor_input(frame, model_shape, region) tensor_input = create_tensor_input(frame, model_shape, region)
scale = float(region[2] - region[0]) / model_shape[0]
detections = [] detections = []
region_detections = object_detector.detect(tensor_input) region_detections = object_detector.detect(tensor_input)
for d in region_detections: for d in region_detections:
box = d[2] box = d[2]
size = region[2] - region[0] size = region[2] - region[0]
x_min = int((box[1]) + region[0]) x_min = int(max(0, box[1]) * scale + region[0])
y_min = int((box[0]) + region[1]) y_min = int(max(0, box[0]) * scale + region[1])
x_max = int((box[3]) + region[0]) x_max = int(min(frame.shape[1], box[3]) * scale + region[0])
y_max = int((box[2]) + region[1]) y_max = int(min(frame.shape[0], box[2]) * scale + region[1])
det = ( det = (
d[0], d[0],
d[1], d[1],