SHOW:
|
|
- or go back to the newest paste.
| 1 | ######################################################################################################################### | |
| 2 | ### ### | |
| 3 | ### RemoveDirtMC ### | |
| 4 | - | ### v1.44 ### |
| 4 | + | ### v1.45 ### |
| 5 | ### **SPECIAL EDITION** ### | |
| 6 | ### By NEPHILIS ### | |
| 7 | ### @2011 ### | |
| 8 | ### mod by Great Dragon to work with NLMeansCL2 ### | |
| 9 | ### mod2 by A.SONY to make it more friendly in motions ### | |
| 10 | ######################################################################################################################### | |
| 11 | ||
| 12 | ################################################################## | |
| 13 | ## ## | |
| 14 | ## -------------- ## | |
| 15 | ## Requirements : - MVTools (v2.5.11.2) ## | |
| 16 | ## -------------- ## | |
| 17 | ## - RemoveGrainT (v1.0) package ## | |
| 18 | ## - RemoveDirt (v0.9) package ## | |
| 19 | ## - AvsRecursion (v0.2) ## | |
| 20 | ## - FluxSmooth (v1.1) ## | |
| 21 | ## ## | |
| 22 | ## As Additional ## | |
| 23 | ## -------- ## | |
| 24 | ## => To Prefiltering with FFT3DFilter ## | |
| 25 | ## - FFT3DFilter (v2.1.1) ## | |
| 26 | ## - Dither_Luma_Rebuild from SMDegrain or elsewhere ## | |
| 27 | ## -------- ## | |
| 28 | ## => To Prefiltering with NLMeansCL2 ## | |
| 29 | ## - NLMeansCL2.dll v0.4.0 (2015-01-08) ## | |
| 30 | ## ## | |
| 31 | ################################################################## | |
| 32 | ||
| 33 | ########################################################################################################################## | |
| 34 | # ------- # | |
| 35 | # Usage : # | |
| 36 | # ------- # | |
| 37 | # # | |
| 38 | # RemoveDirtMC_SE(clip input, bool"GPU", int"radius", int"repmode", int"clmode", bool"TwoPass", int"thSAD", int"thSADC") # | |
| 39 | # # | |
| 40 | ########################################################################################################################## | |
| 41 | ||
| 42 | ################## | |
| 43 | # ----------- # | |
| 44 | # PARAMETERS: # | |
| 45 | # ----------- # | |
| 46 | ################## | |
| 47 | ||
| 48 | ########################################################################################################################## | |
| 49 | # | |
| 50 | # ---------- | |
| 51 | # GPU : -- To choose NLMeansCL or FFT3Dfilter for PreFiltering process. | |
| 52 | # ---------- ATI GPU owners should use "GPU=false" | |
| 53 | # | |
| 54 | # ---------- | |
| 55 | # Radius : -- Temporal Radius for MDeGrain at second stage DeNoising. | |
| 56 | # ---------- | |
| 57 | # | |
| 58 | # ---------- | |
| 59 | # repmode : -- Repair mode value in main RemoveDirtSE function. | |
| 60 | # ---------- | |
| 61 | # | |
| 62 | # ---------- | |
| 63 | # clmode : -- RemoveGrain mode value in main RemoveDirtSE function. | |
| 64 | # ---------- | |
| 65 | # | |
| 66 | # ---------- | |
| 67 | # Twopass : -- To make a Temporal Denoising (MDeGrain) after Spatial Denoising (RemoveDirtSE). | |
| 68 | # ---------- | |
| 69 | # | |
| 70 | # ---------- | |
| 71 | # thSAD, | |
| 72 | # thSADC : -- MDeGrain's thSAD and thSADC parameters. | |
| 73 | # ---------- | |
| 74 | # | |
| 75 | ########################################################################################################################## | |
| 76 | ||
| 77 | Function RemoveDirtMC_SE(clip input, bool"GPU", int"radius", int"repmode", int"clmode", bool"TwoPass", int"thSAD", int"thSADC",float "rStr",float "Amp", bool "luma_rebuild", clip "PreNR") | |
| 78 | ||
| 79 | {
| |
| 80 | ||
| 81 | - | version = "v1.44" |
| 81 | + | version = "v1.45" |
| 82 | ||
| 83 | GPU = default(GPU, true) | |
| 84 | radius = default(radius, 2) | |
| 85 | repmode = default(repmode, 16) | |
| 86 | clmode = default(clmode, 5) | |
| 87 | TwoPass = default(TwoPass, true) | |
| 88 | thSAD = default(thSAD , 150) | |
| 89 | thSADC = default(thSADC, thSAD) | |
| 90 | rStr = default( rStr, 1.0 ) | |
| 91 | Amp = default( Amp, 0.0625 ) | |
| 92 | luma_rebuild = default(luma_rebuild, true) | |
| 93 | ||
| 94 | ox = input.width() | |
| 95 | oy = input.height() | |
| 96 | HD = (ox== 1920 || oy>= 800) ? true : false | |
| 97 | blksize = (HD==true) ? 16:8 | |
| 98 | search = (HD==true) ? 4:5 | |
| 99 | overlap = blksize/2 | |
| 100 | ||
| 101 | src = input | |
| 102 | PreNR = defined(PreNR) ? PreNR : (GPU==true) ? src.NLMeansCL2(h=radius*1.5,u=3,v=3).TemporalDeGrain_SE(5,9) \ | |
| 103 | : src.FFT3DFilter(sigma=radius*2,plane=4,bw=16,bh=16,ow=8,oh=8,bt=3,wintype=1).TemporalDeGrain_SE(5,9) | |
| 104 | - | PreNR = luma_rebuild ? isyv12(input) ? PreNR.Interleaved2Planar().dither_luma_rebuild(S0=rStr,c=Amp, Planar=true).Planar2Interleaved() \ |
| 104 | + | PreNR = luma_rebuild ? isyuy2(input) ? PreNR.Interleaved2Planar().dither_luma_rebuild(S0=rStr,c=Amp).Planar2Interleaved() \ |
| 105 | : PreNR.dither_luma_rebuild(S0=rStr,c=Amp) : PreNR | |
| 106 | PreNR_super = MSuper(PreNR,hpad=4,vpad=4,pel=2,sharp=2) | |
| 107 | src_super = MSuper(src ,hpad=4,vpad=4,pel=2,sharp=2,levels=1) | |
| 108 | ||
| 109 | bwv1 = MAnalyse(PreNR_super, blksize=blksize, search=search, isb=true, delta=1, overlap=overlap, truemotion=true) | |
| 110 | bwv2 = (radius>=2) ? \ | |
| 111 | MAnalyse(PreNR_super, blksize=blksize, search=search, isb=true, delta=2, overlap=overlap, truemotion=true) : BlankClip() | |
| 112 | bwv3 = (radius==3) ? \ | |
| 113 | MAnalyse(PreNR_super, blksize=blksize, search=search, isb=true, delta=3, overlap=overlap, truemotion=true) : BlankClip() | |
| 114 | fwv1 = MAnalyse(PreNR_super, blksize=blksize, search=search, isb=false, delta=1, overlap=overlap, truemotion=true) | |
| 115 | fwv2 = (radius>=2) ? \ | |
| 116 | MAnalyse(PreNR_super, blksize=blksize, search=search, isb=false, delta=2, overlap=overlap, truemotion=true) : BlankClip() | |
| 117 | fwv3 = (radius==3) ? \ | |
| 118 | MAnalyse(PreNR_super, blksize=blksize, search=search, isb=false, delta=3, overlap=overlap, truemotion=true) : BlankClip() | |
| 119 | ||
| 120 | bcomp = MCompensate(src,src_super,bwv1) | |
| 121 | fcomp = MCompensate(src,src_super,fwv1) | |
| 122 | ||
| 123 | NR1 = src.RemoveDirtSE(fcomp, bcomp, repmode=repmode, clmode=clmode) | |
| 124 | NR1_super = ( TwoPass==true ) ? MSuper(NR1,hpad=4,vpad=4,pel=2,sharp=2,levels=1) : BlankClip() | |
| 125 | NR2 = ( TwoPass==false ) ? NR1 | |
| 126 | \ : ( radius==1 ) ? MDeGrain1(NR1, NR1_super, bwv1, fwv1, thSAD=thSAD, thSADC=thSADC) | |
| 127 | \ : ( radius==2 ) ? MDeGrain2(NR1, NR1_super, bwv1, bwv2, fwv1, fwv2, thSAD=thSAD, thSADC=thSADC) | |
| 128 | \ : MDeGrain3(NR1, NR1_super, bwv1, bwv2, bwv3, fwv1, fwv2, fwv3, thSAD=thSAD, thSADC=thSADC) | |
| 129 | Return(NR2) | |
| 130 | ||
| 131 | } | |
| 132 | ||
| 133 | function TemporalDeGrain_SE(clip input, int "mode", int "thr") | |
| 134 | ||
| 135 | {
| |
| 136 | input = isYUY2(input) ? input.Interleaved2Planar() : input | |
| 137 | mode = default(mode, 5) | |
| 138 | thr = default(thr, 8) | |
| 139 | rg = RemoveGrain(input, mode=mode, Planar=true) | |
| 140 | tr = TemporalRepair(rg, input, Planar=true) | |
| 141 | tr = isYUY2(input) ? tr.Planar2Interleaved() : tr | |
| 142 | fs = FluxSmoothT(tr,thr) | |
| 143 | Return(fs) | |
| 144 | ||
| 145 | } | |
| 146 | ||
| 147 | function RemoveDirtSE(clip input, clip fcomp, clip bcomp, int "repmode", int "clmode") | |
| 148 | ||
| 149 | {
| |
| 150 | input = isYUY2(input) ? input.Interleaved2Planar() : input | |
| 151 | fcomp = isYUY2(input) ? fcomp.Interleaved2Planar() : fcomp | |
| 152 | bcomp = isYUY2(input) ? bcomp.Interleaved2Planar() : bcomp | |
| 153 | ||
| 154 | repmode = default(repmode, 16) | |
| 155 | clmode = default(clmode, 5) | |
| 156 | clensed = interleave(fcomp,input,bcomp).Clense(cache=-1, Planar=true).selectevery(3,1) | |
| 157 | clensedn2 = Clense(input, cache=-1, Planar=true) | |
| 158 | sbegin = ForwardClense(input, cache=-1, Planar=true) | |
| 159 | send = BackwardClense(input, cache=-1, Planar=true) | |
| 160 | alt = Repair(SCSelect(input, sbegin, send, clensed, debug=false, Planar=true), input, mode=repmode, Planar=true) | |
| 161 | restore = Repair(clensed, input, mode=repmode, Planar=true) | |
| 162 | corrected = RestoreMotionBlocks(clensed, restore, neighbour=input, neighbour2=clensedn2, alternative=alt, gmthreshold=50, \ | |
| 163 | dist=1, dmode=2, debug=false, noise=8, noisy=12, Planar=true) | |
| 164 | postNR0 = clmode == 0 ? corrected : RemoveGrain(corrected, mode=clmode, Planar=true) | |
| 165 | postNR1 = isYUY2(input) ? postNR0.Planar2Interleaved() : postNR0 | |
| 166 | Return(postNR1) | |
| 167 | ||
| 168 | } |
