1580158
/
SD-CN-Animation
mirror of https://github.com/volotat/SD-CN-Animation.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

better histogram matching. Issue #57

pull/132/head
Alexey Borsky 2023-05-14 17:25:17 +03:00
parent a35f446b69
commit e16f728512
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/core/txt2vid.py
View File

@ -154,7 +154,7 @@ def start_process(*args):
processed_frames, _, _, _ = utils.img2img(args_dict) processed_frames, _, _, _ = utils.img2img(args_dict)
processed_frame = np.array(processed_frames[0]) processed_frame = np.array(processed_frames[0])
processed_frame = skimage.exposure.match_histograms(processed_frame, init_frame, channel_axis=None) processed_frame = skimage.exposure.match_histograms(processed_frame, init_frame, channel_axis=-1)
processed_frame = np.clip(processed_frame, 0, 255).astype(np.uint8) processed_frame = np.clip(processed_frame, 0, 255).astype(np.uint8)
args_dict['mode'] = 0 args_dict['mode'] = 0
@ -166,7 +166,7 @@ def start_process(*args):
#utils.set_CNs_input_image(args_dict, Image.fromarray(curr_frame)) #utils.set_CNs_input_image(args_dict, Image.fromarray(curr_frame))
processed_frames, _, _, _ = utils.img2img(args_dict) processed_frames, _, _, _ = utils.img2img(args_dict)
processed_frame = np.array(processed_frames[0]) processed_frame = np.array(processed_frames[0])
processed_frame = skimage.exposure.match_histograms(processed_frame, init_frame, channel_axis=None) processed_frame = skimage.exposure.match_histograms(processed_frame, init_frame, channel_axis=-1)
processed_frame = np.clip(processed_frame, 0, 255).astype(np.uint8) processed_frame = np.clip(processed_frame, 0, 255).astype(np.uint8)
output_video.write(cv2.cvtColor(processed_frame, cv2.COLOR_RGB2BGR)) output_video.write(cv2.cvtColor(processed_frame, cv2.COLOR_RGB2BGR))