Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Make pyannote.audio optional (still mandatory to run default pipeline) #61

Merged
merged 9 commits into from
Jun 9, 2022

Conversation

juanmc2005
Copy link
Owner

This PR addresses issue #43

Changelog

  • Replace --gpu with --cpu. Runs on GPU by default.
  • Add SegmentationModel and EmbeddingModel as base classes for custom pre-trained models (into a new models module)
    • pyannote models can be loaded using SegmentationModel.from_pyannote("pyannote/segmentation") if pyannote.audio is installed
  • Run diart.stream and diart.benchmark on GPU by default
  • Add audio module and AudioLoader to load audio from a file
  • Merge ChunkLoader into AudioLoader
  • Rename FramewiseModel to SpeakerSegmentation
  • Rename ChunkwiseModel to SpeakerEmbedding
  • Rewrite Binarize so it does not depend on pyannote.audio
  • Unify pipeline configuration so that PipelineConfig is the only source of truth
  • Add some docstrings
  • Update dependencies

@juanmc2005 juanmc2005 added documentation Improvements or additions to documentation feature New feature or request API Improvements to the API labels Jun 8, 2022
@juanmc2005 juanmc2005 added this to the Version 0.4 milestone Jun 8, 2022
@juanmc2005 juanmc2005 self-assigned this Jun 8, 2022
@juanmc2005 juanmc2005 merged commit d6111e6 into develop Jun 9, 2022
@juanmc2005 juanmc2005 deleted the feat/decouple branch June 9, 2022 12:17
@juanmc2005 juanmc2005 mentioned this pull request Jun 20, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
API Improvements to the API documentation Improvements or additions to documentation feature New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant