Advertisement
MonoS

Script for BM3D artifact

Nov 6th, 2016
181
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.80 KB | None | 0 0
  1. import vapoursynth as vs
  2. import nnedi3_resample as edi
  3.  
  4. core = vs.get_core()
  5.  
  6. src = core.ffms2.Source("C:/U.S. Theatrical Trailer (Red Band).m2ts23.mkv")
  7.  
  8. blksize=16
  9. overlap=8
  10. pad = blksize + overlap
  11.  
  12. deint = core.fmtc.bitdepth(src, bits=16).std.CropRel(left=250, right=250)
  13.  
  14. deint = core.fmtc.resample(deint, deint.width+pad, deint.height+pad, sw=deint.width+pad, sh=deint.height+pad, kernel="point")
  15.  
  16. RGB = edi.nnedi3_resample(deint, deint.width, deint.height,csp=vs.RGB48, fast=False)
  17.  
  18. OPPF = core.bm3d.RGB2OPP(RGB, 1)
  19. OPP = core.fmtc.bitdepth(OPPF, bits=16, flt=False, dmode=1, fulls=True, fulld=True)
  20.  
  21. super = core.mv.Super(OPP)
  22.  
  23. bvec2 = core.mv.Analyse(super, isb = True, delta = 2, blksize=blksize, overlap=overlap, truemotion=False)
  24. bvec1 = core.mv.Analyse(super, isb = True, delta = 1, blksize=blksize, overlap=overlap, truemotion=False)
  25. fvec1 = core.mv.Analyse(super, isb = False, delta = 1, blksize=blksize, overlap=overlap, truemotion=False)
  26. fvec2 = core.mv.Analyse(super, isb = False, delta = 2, blksize=blksize, overlap=overlap, truemotion=False)
  27.  
  28. ref = core.mv.Degrain2(OPP, super, bvec1,fvec1,bvec2,fvec2, 425)
  29. ref = core.fmtc.bitdepth(ref, flt=True)
  30.  
  31.  
  32. flt = core.bm3d.VFinal(OPPF, ref, radius=1, matrix=100, sigma=[11,7,7]).bm3d.VAggregate(radius=1, sample=1)
  33. """
  34. denf = core.bm3d.OPP2RGB(flt, sample=1)
  35.  
  36. den = core.fmtc.bitdepth(denf, bits=16, flt=False, dmode=0)
  37.  
  38. den = edi.nnedi3_resample(den[0:55], 946, 720, matd="709", fulld=False, csp=vs.YUV444P16, fulls=True, src_width=1420, src_height=1080, curves="709", curved="709", mats="RGB", sigmoid=True, kernel="spline64") + edi.nnedi3_resample(den[55::], 946, 720, matd="709", fulld=False, csp=vs.YUV444P16, fulls=True, src_width=1420, src_height=1080, curves="709", curved="709", mats="RGB", sigmoid=True, invks=True)
  39. """
  40. flt.set_output()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement