administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-02 09:15:22 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Update to use cv2 to get image and respect args

Browse Source
This commit is contained in:
Nick Mowen 2022-02-15 22:17:21 -07:00
parent 76ad4494f9
commit 53f3064c6f
1 changed files with 19 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
frigate/http.py
View File

@ -356,6 +356,16 @@ def stats():
@bp.route("/<camera_name>/<label>/latest.jpg") @bp.route("/<camera_name>/<label>/latest.jpg")
def latest(camera_name, label): def latest(camera_name, label):
png_bytes = None png_bytes = None
draw_options = {
"bounding_boxes": request.args.get("bbox", type=int),
"timestamp": request.args.get("timestamp", type=int),
"zones": request.args.get("zones", type=int),
"mask": request.args.get("mask", type=int),
"motion_boxes": request.args.get("motion", type=int),
"regions": request.args.get("regions", type=int),
}
resize_quality = request.args.get("quality", default=70, type=int)
if camera_name in current_app.frigate_config.cameras: if camera_name in current_app.frigate_config.cameras:
event_query = ( event_query = (
Event.select() Event.select()
@ -373,13 +383,16 @@ def latest(camera_name, label):
return "No event for {} was found".format(label), 404 return "No event for {} was found".format(label), 404
# read snapshot from disk # read snapshot from disk
with open( frame = cv2.imread(os.path.join(CLIPS_DIR, f"{event.camera}-{event.id}-clean.png"))
os.path.join(CLIPS_DIR, f"{event.camera}-{event.id}-clean.png"), "rb" height = int(request.args.get("h", str(frame.shape[0])))
) as image_file: width = int(height * frame.shape[1] / frame.shape[0])
png_bytes = image_file.read() frame = cv2.resize(frame, dsize=(width, height), interpolation=cv2.INTER_AREA)
ret, jpg = cv2.imencode(
".jpg", frame, [int(cv2.IMWRITE_JPEG_QUALITY), resize_quality]
)
response = make_response(png_bytes) response = make_response(jpg.tobytes())
response.headers["Content-Type"] = "image/png" response.headers["Content-Type"] = "image/jpeg"
return response return response
else: else:
return "Camera named {} not found".format(camera_name), 404 return "Camera named {} not found".format(camera_name), 404