github/DeepFaceLive
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLive synced 2025-08-14 10:47:00 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

upd BackendBase.py

Browse source
This commit is contained in:
iperov 2021-09-30 18:20:24 +04:00
commit 932edfe875
1 changed files with 14 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

16
apps/DeepFaceLive/backend/BackendBase.py
View file

@ -1,6 +1,6 @@
import multiprocessing
import pickle
from typing import List, Union
from typing import List, Union, Tuple
import numpy as np
from xlib import mp as lib_mp
@ -62,7 +62,19 @@ class BackendConnectionData:
"""
self.set_file(name, image.data)
self._weak_heap_image_infos[name] = (image.shape, image.dtype)
def get_image_shape_dtype(self, name:str) -> Union[None, Tuple[List, np.dtype]]:
"""
returns (image shape, dtype) or (None, None) if file does not exist
"""
if name is None:
return None
image_info = self._weak_heap_image_infos.get(name, None)
if image_info is not None:
shape, dtype = image_info
return shape, dtype
return (None, None)
def get_image(self, name : str) -> Union[np.ndarray, None]:
if name is None:
return None