85 lines
2.2 KiB
Python
85 lines
2.2 KiB
Python
|
"""
|
||
|
|
Support for the demo image processing.
|
||
|
|
|
||
|
|
For more details about this component, please refer to the documentation at
|
||
|
|
https://home-assistant.io/components/demo/
|
||
|
|
"""
|
||
|
|
|
||
|
|
from homeassistant.components.image_processing import ImageProcessingEntity
|
||
|
|
from homeassistant.components.image_processing.openalpr_local import (
|
||
|
|
ImageProcessingAlprEntity)
|
||
|
|
|
||
|
|
|
||
|
|
def setup_platform(hass, config, add_devices, discovery_info=None):
|
||
|
|
"""Setup the demo image_processing platform."""
|
||
|
|
add_devices([
|
||
|
|
DemoImageProcessing('camera.demo_camera', "Demo"),
|
||
|
|
DemoImageProcessingAlpr('camera.demo_camera', "Demo Alpr")
|
||
|
|
])
|
||
|
|
|
||
|
|
|
||
|
|
class DemoImageProcessing(ImageProcessingEntity):
|
||
|
|
"""Demo alpr image processing entity."""
|
||
|
|
|
||
|
|
def __init__(self, camera_entity, name):
|
||
|
|
"""Initialize demo alpr."""
|
||
|
|
self._name = name
|
||
|
|
self._camera = camera_entity
|
||
|
|
self._count = 0
|
||
|
|
|
||
|
|
@property
|
||
|
|
def camera_entity(self):
|
||
|
|
"""Return camera entity id from process pictures."""
|
||
|
|
return self._camera
|
||
|
|
|
||
|
|
@property
|
||
|
|
def name(self):
|
||
|
|
"""Return the name of the entity."""
|
||
|
|
return self._name
|
||
|
|
|
||
|
|
@property
|
||
|
|
def state(self):
|
||
|
|
"""Return the state of the entity."""
|
||
|
|
return self._count
|
||
|
|
|
||
|
|
def process_image(self, image):
|
||
|
|
"""Process image."""
|
||
|
|
self._count += 1
|
||
|
|
|
||
|
|
|
||
|
|
class DemoImageProcessingAlpr(ImageProcessingAlprEntity):
|
||
|
|
"""Demo alpr image processing entity."""
|
||
|
|
|
||
|
|
def __init__(self, camera_entity, name):
|
||
|
|
"""Initialize demo alpr."""
|
||
|
|
super().__init__()
|
||
|
|
|
||
|
|
self._name = name
|
||
|
|
self._camera = camera_entity
|
||
|
|
|
||
|
|
@property
|
||
|
|
def camera_entity(self):
|
||
|
|
"""Return camera entity id from process pictures."""
|
||
|
|
return self._camera
|
||
|
|
|
||
|
|
@property
|
||
|
|
def confidence(self):
|
||
|
|
"""Return minimum confidence for send events."""
|
||
|
|
return 80
|
||
|
|
|
||
|
|
@property
|
||
|
|
def name(self):
|
||
|
|
"""Return the name of the entity."""
|
||
|
|
return self._name
|
||
|
|
|
||
|
|
def process_image(self, image):
|
||
|
|
"""Process image."""
|
||
|
|
demo_data = {
|
||
|
|
'AC3829': 98.3,
|
||
|
|
'BE392034': 95.5,
|
||
|
|
'CD02394': 93.4,
|
||
|
|
'DF923043': 90.8
|
||
|
|
}
|
||
|
|
|
||
|
|
self.process_plates(demo_data, 1)
|