mexca.container

Containers of pipeline components.

Module Contents

Classes

BaseContainer

Base class for container components. Only for internal use.

FaceExtractorContainer

Container for FaceExtractor component.

SpeakerIdentifierContainer

Container for SpeakerIdentifier component.

VoiceExtractorContainer

Container for VoiceExtractor component.

AudioTranscriberContainer

Container for AudioTrascriber component.

SentimentExtractorContainer

Container for SentimentExtractor component.

class mexca.container.BaseContainer(image_name: str)[source]

Base class for container components. Only for internal use.

class mexca.container.FaceExtractorContainer(num_faces: Optional[int], min_face_size: int = 20, thresholds: Tuple[float] = (0.6, 0.7, 0.7), factor: float = 0.709, post_process: bool = True, select_largest: bool = True, selection_method: Optional[str] = None, keep_all: bool = True, device: Optional[torch.device] = 'cpu', embeddings_model: str = 'vggface2', au_model: str = 'xgb', landmark_model: str = 'mobilefacenet', image_name: str = 'mexca/face-extractor')[source]

Bases: BaseContainer

Container for FaceExtractor component.

Parameters

image_name (str, default='mexca-face-extractor') – Name of the image to create a container from. Pulls the image from Docker Hub if not found locally.

See also

FaceExtractor

class mexca.container.SpeakerIdentifierContainer(num_speakers: Optional[int] = None, use_auth_token: Union[bool, str] = True, image_name: str = 'mexca/speaker-identifier')[source]

Bases: BaseContainer

Container for SpeakerIdentifier component.

Parameters

image_name (str, default='mexca-speaker-identifier') – Name of the image to create a container from. Pulls the image from Docker Hub if not found locally.

See also

SpeakerIdentifier

class mexca.container.VoiceExtractorContainer(image_name: str = 'mexca/voice-extractor')[source]

Bases: BaseContainer

Container for VoiceExtractor component.

Parameters

image_name (str, default='mexca-voice-extractor') – Name of the image to create a container from. Pulls the image from Docker Hub if not found locally.

See also

VoiceExtractor

class mexca.container.AudioTranscriberContainer(whisper_model: Optional[str] = 'small', device: Optional[Union[str, torch.device]] = 'cpu', sentence_rule: Optional[str] = None, image_name: str = 'mexca/audio-transcriber')[source]

Bases: BaseContainer

Container for AudioTrascriber component.

Parameters

image_name (str, default='mexca-audio-transcriber') – Name of the image to create a container from. Pulls the image from Docker Hub if not found locally.

See also

AudioTranscriber

class mexca.container.SentimentExtractorContainer(image_name: str = 'mexca/sentiment-extractor')[source]

Bases: BaseContainer

Container for SentimentExtractor component.

Parameters

image_name (str, default='mexca-sentiment-extractor') – Name of the image to create a container from. Pulls the image from Docker Hub if not found locally.

See also

SentimentExtractor