github/DeepFaceLab
1
0
Fork 0
mirror of https://github.com/iperov/DeepFaceLab.git synced 2025-07-05 20:42:11 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

upd cv2ex

Browse source
This commit is contained in:
Colombo 2020-07-03 19:30:52 +04:00
parent 3283dce96a
commit b0ad36de94
1 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

9
core/cv2ex.py
View file

@ -2,6 +2,7 @@ import cv2
import numpy as np import numpy as np
from pathlib import Path from pathlib import Path
from core.interact import interact as io from core.interact import interact as io
from core import imagelib
import traceback import traceback
def cv2_imread(filename, flags=cv2.IMREAD_UNCHANGED, loader_func=None, verbose=True): def cv2_imread(filename, flags=cv2.IMREAD_UNCHANGED, loader_func=None, verbose=True):
@ -29,3 +30,11 @@ def cv2_imwrite(filename, img, *args):
stream.write( buf ) stream.write( buf )
except: except:
pass pass
def cv2_resize(x, *args, **kwargs):
h,w,c = x.shape
x = cv2.resize(x, *args, **kwargs)
x = imagelib.normalize_channels(x, c)
return x