mirrors/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove deoldify

Browse Source
This commit is contained in:
henryruhs 2024-04-02 01:15:09 +02:00
parent af97de90e2
commit bd8034cbc7
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
facefusion/processors/frame/modules/frame_colorizer.py
View File

@ -29,14 +29,12 @@ MODELS : ModelSet =\
{ {
'ddcolor': 'ddcolor':
{ {
'type': 'ddcolor',
'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models/ddcolor.onnx', 'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models/ddcolor.onnx',
'path': resolve_relative_path('../.assets/models/ddcolor.onnx'), 'path': resolve_relative_path('../.assets/models/ddcolor.onnx'),
'size': (512, 512) 'size': (512, 512)
}, },
'ddcolor_artistic': 'ddcolor_artistic':
{ {
'type': 'ddcolor',
'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models/ddcolor_artistic.onnx', 'url': 'https://github.com/facefusion/facefusion-assets/releases/download/models/ddcolor_artistic.onnx',
'path': resolve_relative_path('../.assets/models/ddcolor_artistic.onnx'), 'path': resolve_relative_path('../.assets/models/ddcolor_artistic.onnx'),
'size': (512, 512) 'size': (512, 512)
@ -153,7 +151,7 @@ def prepare_temp_frame(temp_vision_frame : VisionFrame) -> VisionFrame:
temp_vision_frame = (temp_vision_frame / 255.0).astype(numpy.float32) temp_vision_frame = (temp_vision_frame / 255.0).astype(numpy.float32)
temp_vision_frame = cv2.resize(temp_vision_frame, model_size) temp_vision_frame = cv2.resize(temp_vision_frame, model_size)
temp_vision_frame = cv2.cvtColor(temp_vision_frame, cv2.COLOR_BGR2Lab)[:, :, :1] temp_vision_frame = cv2.cvtColor(temp_vision_frame, cv2.COLOR_BGR2Lab)[:, :, :1]
temp_vision_frame = numpy.concatenate((temp_vision_frame, numpy.zeros_like(temp_vision_frame), numpy.zeros_like(temp_vision_frame)), axis=-1) temp_vision_frame = numpy.concatenate((temp_vision_frame, numpy.zeros_like(temp_vision_frame), numpy.zeros_like(temp_vision_frame)), axis = -1)
temp_vision_frame = cv2.cvtColor(temp_vision_frame, cv2.COLOR_LAB2RGB) temp_vision_frame = cv2.cvtColor(temp_vision_frame, cv2.COLOR_LAB2RGB)
temp_vision_frame = cv2.resize(temp_vision_frame, model_size) temp_vision_frame = cv2.resize(temp_vision_frame, model_size)
temp_vision_frame = temp_vision_frame.transpose((2, 0, 1)) temp_vision_frame = temp_vision_frame.transpose((2, 0, 1))