cleanup imports

This commit is contained in:
Blake Blackshear 2020-01-04 12:00:29 -06:00
parent a43fd96349
commit 5d0c12fbd4
3 changed files with 7 additions and 6 deletions

View File

@ -5,7 +5,8 @@ import threading
import prctl import prctl
import numpy as np import numpy as np
from edgetpu.detection.engine import DetectionEngine 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): class PreppedQueueProcessor(threading.Thread):
def __init__(self, cameras, prepped_frame_queue, fps, queue_full): def __init__(self, cameras, prepped_frame_queue, fps, queue_full):

View File

@ -6,7 +6,7 @@ import prctl
import itertools import itertools
import numpy as np import numpy as np
from scipy.spatial import distance as dist 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): class ObjectCleaner(threading.Thread):
def __init__(self, objects_parsed, detected_objects): def __init__(self, objects_parsed, detected_objects):

View File

@ -11,10 +11,10 @@ import numpy as np
import prctl import prctl
import itertools import itertools
from collections import defaultdict from collections import defaultdict
from . util import tonumpyarray, LABELS, draw_box_with_label, calculate_region, EventsPerSecond from frigate.util import tonumpyarray, LABELS, draw_box_with_label, calculate_region, EventsPerSecond
from . object_detection import RegionPrepper, RegionRequester from frigate.object_detection import RegionPrepper, RegionRequester
from . objects import ObjectCleaner, BestFrames, DetectedObjectsProcessor, RegionRefiner, ObjectTracker from frigate.objects import ObjectCleaner, BestFrames, DetectedObjectsProcessor, RegionRefiner, ObjectTracker
from . mqtt import MqttObjectPublisher from frigate.mqtt import MqttObjectPublisher
# Stores 2 seconds worth of frames so they can be used for other threads # Stores 2 seconds worth of frames so they can be used for other threads
class FrameTracker(threading.Thread): class FrameTracker(threading.Thread):