cutcutcodec.core.analysis.video.metric.uvq_google.distortionnet.DistortionNetInference

class cutcutcodec.core.analysis.video.metric.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.

get_features_for_all_frames(video: ndarray) → tuple[ndarray, ndarray][source]

Eval the model.

predict_and_get_features(frame) → tuple[ndarray, ndarray][source]

Eval the model.