5
0

yt_common: upstream sraa changes

This commit is contained in:
louis f 2021-05-31 18:17:42 -04:00
parent f9816d31c2
commit 362a8672d1
Signed by: louis
GPG Key ID: 44D7E1DE4E23D6F2
3 changed files with 29 additions and 3 deletions

View File

@ -0,0 +1 @@
../yt_common/mypy.ini

View File

@ -0,0 +1,25 @@
#!/usr/bin/env python3
import setuptools
name = "tensura_common"
version = "0.0.0"
release = "0.0.0"
setuptools.setup(
name=name,
version=release,
author="louis",
author_email="louis@poweris.moe",
description="tensura common module",
packages=["tensura_common"],
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
],
package_data={
'pripri_common': ['py.typed', 'final-settings'],
},
python_requires='>=3.8',
)

View File

@ -1,6 +1,6 @@
import vapoursynth as vs
from yt_common.antialiasing import mask_strong, sraa_clamp, supersample_aa
from yt_common.antialiasing import mask_strong, sraa_clamp
from yt_common.data import FSRCNNX
from yt_common.scale import nnedi3_double
@ -72,11 +72,11 @@ def deband(clip: vs.VideoNode, strong: Optional[List[Range]] = None,
def antialias(clip: vs.VideoNode, strong: List[Range],
sangnom: Optional[List[Tuple[Range, List[BoundingBox]]]] = None) -> vs.VideoNode:
clamp = sraa_clamp(clip, mask=mask_strong)
sraa_13 = upscaled_sraa(clip, rfactor=1.3, downscaler=Bicubic(b=0, c=1/2).scale)
sraa_13 = upscaled_sraa(clip, rfactor=1.3)
sraa_13 = core.std.MaskedMerge(clip, sraa_13, mask_strong(sraa_13))
clamp = replace_ranges(clamp, sraa_13, strong)
if sangnom:
sn = supersample_aa(sraa_13, rfactor=1.5, aafun=lambda c, s: c.sangnom.SangNom())
sn = upscaled_sraa(sraa_13, aafun=lambda c: c.sangnom.SangNom())
sn = core.std.MaskedMerge(sraa_13, sn, mask_strong(sn))
for r, ms in sangnom:
mask = core.std.BlankClip(clip.std.ShufflePlanes(planes=0, colorfamily=vs.GRAY))