thebears/inference_rpi_hailo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

YACWC

Browse Source
This commit is contained in:
The Bears
2025-11-06 15:22:07 -05:00
parent 47871c5370
commit e03cc13d2b
4 changed files with 421 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
utils.py
View File

@@ -58,11 +58,11 @@ def get_snap(url,
'cmd': 'Snap',
'channel': 0,
'rs': ''.join(choices(string.ascii_uppercase + string.digits, k=10)),
'snapType': 'sub',
'snapType': 'main',
'user': username,
'password': password,
}
if width is not None and height is not None:
if width is not None and height is not None and data['snapType'] == 'sub':
data['width'] = width
data['height'] = height
parms = parse.urlencode(data, safe="!").encode("utf-8")
@@ -136,22 +136,25 @@ def kafka_poster(kafka_results_queue):
while True:
msg = kafka_results_queue.get()
msg_type = msg.get('type', 'n/a')
if msg_type in {'objdet'}:
data = msg['data']
cam_name = data['camera_name']
topic_name = f'streaming_{cam_name}_{msg_type}'
kwq.kafka_client.send(topic=topic_name,
key=data['timestamp'],
value=data)
elif msg_type in {'image'}:
data = msg['data']
cam_name = data['camera_name']
topic_name = f'streaming_{cam_name}_{msg_type}'
kwq.kafka_client.send(topic=topic_name,
key=data['timestamp'],
value=data['image_bytes'],
serialize_value=False)
try:
if msg_type in {'objdet'}:
data = msg['data']
cam_name = data['camera_name']
topic_name = f'streaming_{cam_name}_{msg_type}'
kwq.kafka_client.send(topic=topic_name,
key=data['timestamp'],
value=data)
elif msg_type in {'image'}:
data = msg['data']
cam_name = data['camera_name']
topic_name = f'streaming_{cam_name}_{msg_type}'
kwq.kafka_client.send(topic=topic_name,
key=data['timestamp'],
value=data['image_bytes'],
serialize_value=False)
except Exception as e:
logger.error(f'Error when publishing to {topic_name}: {str(e)}')
def model_scoring_callback(completion_info, bindings, data, kafka_queue):
if completion_info.exception: