mirror of
https://github.com/iperov/DeepFaceLab.git
synced 2025-07-06 13:02:15 -07:00
50f892d57d
If you want, you can manually remove unnecessary angles from src faceset after sort by yaw. Optimized sample generators (CPU workers). Now they consume less amount of RAM and work faster. added 4.2.other) data_src/dst util faceset pack.bat Packs /aligned/ samples into one /aligned/samples.pak file. After that, all faces will be deleted. 4.2.other) data_src/dst util faceset unpack.bat unpacks faces from /aligned/samples.pak to /aligned/ dir. After that, samples.pak will be deleted. Packed faceset load and work faster.
25 lines
771 B
Python
25 lines
771 B
Python
import cv2
|
|
import numpy as np
|
|
from pathlib import Path
|
|
|
|
#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:
|
|
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
|