cutcutcodec.core.analysis.video.quality.uvq_google.compressionnet.CompressionNetInference¶

class cutcutcodec.core.analysis.video.quality.uvq_google.compressionnet.CompressionNetInference(eval_mode=True, video_height=720, video_width=1280, video_channels=3, patch_height=180, patch_width=320, depth=5, feature_channels=100, feature_height=16, feature_width=16, label_dim=1, **kwargs)[source]¶

Give a score about compression artifact.

get_features_for_all_frames(video: ndarray) → tuple[ndarray, ndarray][source]¶: Eval the model on all frames.

predict_and_get_features(patch) → tuple[ndarray, ndarray][source]¶: Eval the model on one frame.