public/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-12-10 09:33:58 +09:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

fix(deps): update machine-learning (#10740)

Browse Source 
* fix(deps): update machine-learning

* update openvino options, cuda

* update openvino build

* fix indentation

* update minimum nvidia driver

---------

Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com>
Co-authored-by: mertalev <101130780+mertalev@users.noreply.github.com>
This commit is contained in:
renovate[bot]
2024-07-21 19:30:24 -04:00
committed by GitHub
parent 8b773a2b2e
commit b53bd8c525
8 changed files with 323 additions and 284 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
machine-learning/app/models/transforms.py
View File

@@ -47,8 +47,8 @@ def pil_to_cv2(image: Image.Image) -> NDArray[np.uint8]:
def decode_pil(image_bytes: bytes | IO[bytes] | Image.Image) -> Image.Image:
if isinstance(image_bytes, Image.Image):
return image_bytes
image = Image.open(BytesIO(image_bytes) if isinstance(image_bytes, bytes) else image_bytes)
image.load() # type: ignore
image: Image.Image = Image.open(BytesIO(image_bytes) if isinstance(image_bytes, bytes) else image_bytes)
image.load()
if not image.mode == "RGB":
image = image.convert("RGB")
return image