From f5a2252b29c3c39c3a14dd32446feeb1abf820a0 Mon Sep 17 00:00:00 2001 From: Blake Blackshear Date: Sat, 4 Jan 2020 12:00:29 -0600 Subject: [PATCH] cleanup imports --- frigate/object_detection.py | 3 ++- frigate/objects.py | 2 +- frigate/video.py | 8 ++++---- 3 files changed, 7 insertions(+), 6 deletions(-) diff --git a/frigate/object_detection.py b/frigate/object_detection.py index 57e709f64..d3d916267 100644 --- a/frigate/object_detection.py +++ b/frigate/object_detection.py @@ -5,7 +5,8 @@ import threading import prctl import numpy as np from edgetpu.detection.engine import DetectionEngine -from . util import tonumpyarray, LABELS, PATH_TO_CKPT + +from frigate.util import tonumpyarray, LABELS, PATH_TO_CKPT class PreppedQueueProcessor(threading.Thread): def __init__(self, cameras, prepped_frame_queue, fps, queue_full): diff --git a/frigate/objects.py b/frigate/objects.py index 5f7aeba2a..f35274bc1 100644 --- a/frigate/objects.py +++ b/frigate/objects.py @@ -6,7 +6,7 @@ import prctl import itertools import numpy as np from scipy.spatial import distance as dist -from . util import draw_box_with_label, LABELS, compute_intersection_rectangle, compute_intersection_over_union, calculate_region +from frigate.util import draw_box_with_label, LABELS, compute_intersection_rectangle, compute_intersection_over_union, calculate_region class ObjectCleaner(threading.Thread): def __init__(self, objects_parsed, detected_objects): diff --git a/frigate/video.py b/frigate/video.py index 6e22dc13a..08ce987c4 100644 --- a/frigate/video.py +++ b/frigate/video.py @@ -11,10 +11,10 @@ import numpy as np import prctl import itertools from collections import defaultdict -from . util import tonumpyarray, LABELS, draw_box_with_label, calculate_region, EventsPerSecond -from . object_detection import RegionPrepper, RegionRequester -from . objects import ObjectCleaner, BestFrames, DetectedObjectsProcessor, RegionRefiner, ObjectTracker -from . mqtt import MqttObjectPublisher +from frigate.util import tonumpyarray, LABELS, draw_box_with_label, calculate_region, EventsPerSecond +from frigate.object_detection import RegionPrepper, RegionRequester +from frigate.objects import ObjectCleaner, BestFrames, DetectedObjectsProcessor, RegionRefiner, ObjectTracker +from frigate.mqtt import MqttObjectPublisher # Stores 2 seconds worth of frames so they can be used for other threads class FrameTracker(threading.Thread):