urbanlc.model.download.download_model

urbanlc.model.download.download_model(sensor_type: str) → Module

Downloads a pre-trained model from Google Drive.

This function downloads a pre-trained model from Google Drive based on the specified sensor type. The model is stored locally in the ‘pretrained_models’ directory.

Parameters:

sensor_type (str) – Type of sensor for selecting the appropriate pre-trained model.

Returns:

Pre-trained model.

Return type:

torch.nn.Module