mexca.container
Containers of pipeline components.
Module Contents
Classes
Base class for container components. Only for internal use. |
|
Container for FaceExtractor component. |
|
Container for SpeakerIdentifier component. |
|
Container for VoiceExtractor component. |
|
Container for AudioTrascriber component. |
|
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