sc2bench.models.wrapper

Functions

get_wrapped_classification_model(...)

Gets a wrapped image classification model.

register_wrapper_class(cls)

Registers a model wrapper class.

wrap_model(wrapper_model_name, model, ...)

Wraps a model and a compression model with a wrapper module.

Classes

CodecFeatureCompressionClassifier(...[, ...])

A wrapper module for codec feature compression model injected to a classification model.

CodecInputCompressionClassifier(...[, ...])

A wrapper module for codec input compression model followed by a classification model.

EntropicClassifier(classification_model, ...)

A wrapper module for entropic compression model injected to a classification model.

NeuralInputCompressionClassifier(...[, ...])

A wrapper module for neural input compression model followed by a classification model.

SplitClassifier(classification_model, ...[, ...])

A wrapper module for naively splitting a classification model.