#!/usr/bin/python3
import ast
import cv2
import time
from camera_stream import CameraStream
from object_detector import ObjectDetector
config_path = "local.conf"
config_file = open(config_path, "r")
config_dict = ast.literal_eval(config_file.read())
camera_stream = CameraStream(config_dict["camera_stream_path"])
camera_stream.start()
objdet = ObjectDetector(config_dict["detection_classes_path"], config_dict["model_path"])
while(camera_stream.isAlive()):
ret, last_frame = camera_stream.getLastFrame()
if(ret):
cv2.imshow('open_watchman', objdet.detectAndPlot(last_frame, 0.4))
if cv2.waitKey(10) & 0xFF == ord('q'):
break
camera_stream.heyStopItNow()
camera_stream.join()
#!/usr/bin/python3
import ast
import cv2
import numpy as np
from object_detector import ObjectDetector
config_path = "local.conf"
config_file = open(config_path, "r")
config_dict = ast.literal_eval(config_file.read())
objdet = ObjectDetector(config_dict["detection_classes_path"],
config_dict["model_path"])
input_image = cv2.imread("boats_test.jpg")
cv2.imwrite("local_output.jpg", objdet.detectAndPlot(input_image))