【DE_RGBSHIT】用于mpv的反rgb位移的vs脚本

为什么要做这个东西就不赘述了，和ai合作写了这个东西，能一定程度反制rgbshift滤镜。
性能问题大概不会很多，如果有代码上的问题欢迎指正。

1. import vapoursynth as vs
   
2. from vapoursynth import core
   
3. 
   
4. core = vs.core
   
5. clip = video_in
   
6. 
   
7. # ====== 配置参数 ======
   
8. SHIFT_X = -1.0    # 水平位移量（正数=右移）
   
9. SHIFT_Y = -0.5    # 垂直位移量（正数=下移，负数=上移）
   
10. 
    
11. # ====== RGB处理流程 ======
    
12. # 转换为 RGB 格式（假设输入为 BT.709）
    
13. rgb = core.resize.Bilinear(
    
14.     clip,
    
15.     format=vs.RGBS,
    
16.     matrix_in_s="709" if clip.format.color_family == vs.YUV else "rgb"
    
17. )
    
18. 
    
19. # 分离 RGB 通道
    
20. r, g, b = core.std.SplitPlanes(rgb)
    
21. 
    
22. # ====== 蓝色通道位移处理 ======
    
23. # 水平+垂直联合位移
    
24. shifted_b = core.resize.Bilinear(
    
25.     b,
    
26.     src_left   = -SHIFT_X,          # 右移1像素（负值=向右采样）
    
27.     src_top    = -abs(SHIFT_Y),     # 上移0.5像素（负值=向上采样）
    
28.     src_width  = b.width + SHIFT_X, # 扩展采样区域
    
29.     src_height = b.height + abs(SHIFT_Y),
    
30.     width      = b.width,           # 输出尺寸不变
    
31.     height     = b.height,
    
32.     filter_param_a=0, filter_param_b=0.5  # 最近邻插值
    
33. )
    
34. 
    
35. # ====== 重组通道 ======
    
36. shifted_clip = core.std.ShufflePlanes(
    
37.     clips=[r, g, shifted_b],
    
38.     planes=[0, 0, 0],
    
39.     colorfamily=vs.RGB
    
40. )
    
41. 
    
42. # ====== 转回YUV格式 ======
    
43. yuv_clip = core.resize.Bicubic(
    
44.     shifted_clip,
    
45.     format=vs.YUV420P10,
    
46.     matrix_s="709",
    
47.     dither_type="error_diffusion"
    
48. )
    
49. 
    
50. yuv_clip.set_output()
    

复制代码