diff --git a/detect_objects.py b/detect_objects.py index 5518baa4d..42b84e895 100644 --- a/detect_objects.py +++ b/detect_objects.py @@ -37,6 +37,8 @@ REGIONS = os.getenv('REGIONS') DETECTED_OBJECTS = [] +DEBUG = (os.getenv('DEBUG') == '1') + # Loading label map label_map = label_map_util.load_labelmap(PATH_TO_LABELS) categories = label_map_util.convert_label_map_to_categories(label_map, max_num_classes=NUM_CLASSES, @@ -102,7 +104,6 @@ class ObjectParser(threading.Thread): global DETECTED_OBJECTS while True: obj = self._object_queue.get() - print(obj) DETECTED_OBJECTS.append(obj) # notify that objects were parsed @@ -259,7 +260,7 @@ def main(): region['motion_detected'], frame_shape, region['size'], region['x_offset'], region['y_offset'], - False)) + DEBUG)) detection_process.daemon = True detection_processes.append(detection_process) @@ -271,7 +272,7 @@ def main(): frame_shape, region['size'], region['x_offset'], region['y_offset'], region['min_object_size'], region['mask'], - True)) + DEBUG)) motion_process.daemon = True motion_processes.append(motion_process) @@ -405,7 +406,6 @@ def fetch_frames(shared_arr, shared_frame_time, frame_lock, frame_ready, frame_s def process_frames(shared_arr, object_queue, shared_frame_time, frame_lock, frame_ready, motion_detected, frame_shape, region_size, region_x_offset, region_y_offset, debug): - debug = True # shape shared input array into frame for processing arr = tonumpyarray(shared_arr).reshape(frame_shape)