cutcutcodec.core.analysis.video.quality.uvq_google.distortionnet

A modified version of the Google UVQ source file.

As the original file is under apache lisence, I should mention that this is a modified version of the source file:

https://github.com/google/uvq/blob/main/uvq_pytorch/utils/distortionnet.py

Classes

DistortionNet([dropout])

Model to eval the distorsion.

DistortionNetInference([eval_mode, ...])

Model to eval the distortion.

Details

class cutcutcodec.core.analysis.video.quality.uvq_google.distortionnet.DistortionNet(dropout=0.2, **kwargs)[source]

Model to eval the distorsion.

Initialize internal Module state, shared by both nn.Module and ScriptModule.

class cutcutcodec.core.analysis.video.quality.uvq_google.distortionnet.DistortionNetInference(eval_mode=True, video_height=720, video_width=1280, video_channels=3, patch_height=360, patch_width=640, depth=1, feature_channels=100, feature_height=16, feature_width=16, label_dim=26, **kwargs)[source]

Model to eval the distortion.