class RawJpgHsv(NamedTuple):
"""Raw image bytes, the jpeg image and the HSV (hue saturation value) image."""
+
raw: Optional[bytes]
jpg: Optional[np.ndarray]
hsv: Optional[np.ndarray]
class SanityCheckImageMetadata(NamedTuple):
"""Is a Blue Iris image bad (big grey borders around it) or infrared?"""
+
is_bad_image: bool
is_infrared_image: bool
def sanity_check_image(hsv: np.ndarray) -> SanityCheckImageMetadata:
"""See if a Blue Iris or Shinobi image is bad and infrared."""
+
def is_near(a, b) -> bool:
return abs(a - b) < 3
for c in range(cols):
pixel = hsv[(r, c)]
if (
- is_near(pixel[0], 16) and
- is_near(pixel[1], 117) and
- is_near(pixel[2], 196)
+ is_near(pixel[0], 16)
+ and is_near(pixel[1], 117)
+ and is_near(pixel[2], 196)
):
weird_orange_count += 1
- elif (is_near(pixel[0], 0) and is_near(pixel[1], 0)):
+ elif is_near(pixel[0], 0) and is_near(pixel[1], 0):
hs_zero_count += 1
logger.debug(f"hszero#={hs_zero_count}, weird_orange={weird_orange_count}")
return SanityCheckImageMetadata(
- hs_zero_count > (num_pixels * 0.75), weird_orange_count > (num_pixels * 0.75)
+ hs_zero_count > (num_pixels * 0.75),
+ weird_orange_count > (num_pixels * 0.75),
)
def _fetch_camera_image(
camera_name: str, *, width: int = 256, quality: int = 70
) -> RawJpgHsv:
- """Fetch a webcam image given the camera name.
-
- """
+ """Fetch a webcam image given the camera name."""
logger.debug("Trying to fetch camera image from video server")
raw = fetch_camera_image_from_video_server(
camera_name, width=width, quality=quality
)
if raw is None:
- logger.debug(
- "Reading from video server failed; trying direct RTSP stream"
- )
+ logger.debug("Reading from video server failed; trying direct RTSP stream")
raw = fetch_camera_image_from_rtsp_stream(camera_name, width=width)
if raw is not None and len(raw) > 0:
tmp = np.frombuffer(raw, dtype="uint8")
if __name__ == '__main__':
import doctest
+
doctest.testmod()