mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-03-12 20:42:45 -07:00
28 lines
870 B
Python
28 lines
870 B
Python
import cv2
|
|
import numpy as np
|
|
from pathlib import Path
|
|
import traceback
|
|
|
|
#allows to open non-english characters path
|
|
def cv2_imread(filename, flags=cv2.IMREAD_UNCHANGED, loader_func=None):
|
|
try:
|
|
if loader_func is not None:
|
|
bytes = bytearray(loader_func(filename))
|
|
else:
|
|
with open(filename, "rb") as stream:
|
|
bytes = bytearray(stream.read())
|
|
numpyarray = np.asarray(bytes, dtype=np.uint8)
|
|
return cv2.imdecode(numpyarray, flags)
|
|
except:
|
|
io.log_err(f"Exception occured in cv2_imread : {traceback.format_exc()}")
|
|
return None
|
|
|
|
def cv2_imwrite(filename, img, *args):
|
|
ret, buf = cv2.imencode( Path(filename).suffix, img, *args)
|
|
if ret == True:
|
|
try:
|
|
with open(filename, "wb") as stream:
|
|
stream.write( buf )
|
|
except:
|
|
pass
|