想请教下创建自定义脚本移除rgb shift的问题

[zdrzdr]

我的编程语言基础大概就一点点脚本语言，看教程和gpt4辅助写了个脚本，好像一直卡在色域转换这里了。。大佬能不能帮忙看看

[zdrzdr]

### 解决rgb shit问题
import vapoursynth as vs
from vapoursynth import core                                              
from k7sfunc import * 
import numpy as np

core = vs.core

# 替换成你的视频输入
clip = video_in

# 确保输入剪辑是RGB格式
clip = core.resize.Bicubic(clip, format=vs.RGB24)

# RGB通道偏移的反向处理
def reverse_rgb_shift(clip, shift_value):
    def process_frame(n, f):
        frame = f[0]
        
        # 将每个平面转换为NumPy数组
        r = np.frombuffer(frame[0], dtype=np.uint8).reshape((frame.height, frame.width))
        g = np.frombuffer(frame[1], dtype=np.uint8).reshape((frame.height, frame.width))
        b = np.frombuffer(frame[2], dtype=np.uint8).reshape((frame.height, frame.width))
        
        # 将三个平面堆叠成一个三维数组
        frame_array = np.dstack((r, g, b))
        
        # 反向处理蓝色通道的偏移
        frame_array[:, :, 2] = np.roll(frame_array[:, :, 2], -shift_value, axis=1)
        
        # 创建新的帧
        new_frame = frame.copy()
        new_frame[0][:] = frame_array[:, :, 0].flatten()
        new_frame[1][:] = frame_array[:, :, 1].flatten()
        new_frame[2][:] = frame_array[:, :, 2].flatten()
        
        return new_frame
    
    return core.std.ModifyFrame(clip, clips=[clip], selector=process_frame)

# 设置反向RGB shift的偏移值
reverse_shift_value = 10  # 根据需要调整

# 应用反向RGB shift效果
reversed_clip = reverse_rgb_shift(clip, reverse_shift_value)

# 输出结果
reversed_clip.set_output()

[hooke007]

yuv和rgb互转必须要指定matrix 示例

core.resize.Bilinear(clip=xxx, format=vs.RGBH, matrix_in_s="709")
core.resize.Bilinear(clip=xxx, format=vs.YUV420P10, matrix_s="709")

你的最终输出必须是yuv不可以是rgb

[zdrzdr]

[hooke007]

内置功能已存在重复
https://github.com/hooke007/MPV_lazy/wiki/3_K7sfunc#csc_rb