allow setting size and cropping of snapshots and best.jpg endpoint

This commit is contained in:
Blake Blackshear
2020-09-13 08:57:47 -05:00
parent 1ce993051e
commit 50e568b84c
3 changed files with 25 additions and 6 deletions

View File

@@ -381,10 +381,14 @@ def main():
@app.route('/<camera_name>/<label>/best.jpg')
def best(camera_name, label):
if camera_name in CONFIG['cameras']:
best_frame = object_processor.get_best(camera_name, label)
if best_frame is None:
best_frame = np.zeros((720,1280,3), np.uint8)
best_object = object_processor.get_best(camera_name, label)
best_frame = best_object.get('frame', np.zeros((720,1280,3), np.uint8))
crop = bool(request.args.get('crop', 0))
if crop:
region = best_object.get('region', [0,0,300,300])
best_frame = best_frame[region[1]:region[3], region[0]:region[2]]
height = int(request.args.get('h', str(best_frame.shape[0])))
width = int(height*best_frame.shape[1]/best_frame.shape[0])