acies.vehicle_classifier.vfm#

Classes

VibroFM

Vibration Foundation Model.
class acies.vehicle_classifier.vfm.VibroFM#

Bases: Classifier

Vibration Foundation Model.

__init__(modality, classifier_config_file, *args, **kwargs)#

Initialize the VibroFM classifier.

Parameters:

  • modality (str) – The modality to use (e.g., ‘seismic’, ‘audio’).

  • classifier_config_file (pathlib.Path) – The path to the classifier configuration file.

infer(samples)#

Run inference on the provided samples.

Parameters:

samples (dict[str, dict[int, numpy.ndarray]]) – The input samples for inference.

Returns:

The inference results.

Return type:

dict[str, float]

load_model(classifier_config_file)#

Load the model for inference.

Parameters:

classifier_config_file (pathlib.Path) – The path to the classifier configuration file.

Returns:

The loaded model.

Return type:

torch.nn.Module