mirror of
https://github.com/w-okada/voice-changer.git
synced 2025-01-24 05:55:01 +03:00
d5561c2212
- Add Crepe Full/Tiny (onnx) - remove test connect for local Refactor: - RVC: comment out module importer
45 lines
1.9 KiB
Python
45 lines
1.9 KiB
Python
from typing import Protocol
|
|
from const import PitchExtractorType
|
|
from voice_changer.RVC.pitchExtractor.CrepeOnnxPitchExtractor import CrepeOnnxPitchExtractor
|
|
from voice_changer.RVC.pitchExtractor.DioPitchExtractor import DioPitchExtractor
|
|
from voice_changer.RVC.pitchExtractor.HarvestPitchExtractor import HarvestPitchExtractor
|
|
from voice_changer.RVC.pitchExtractor.CrepePitchExtractor import CrepePitchExtractor
|
|
from voice_changer.RVC.pitchExtractor.PitchExtractor import PitchExtractor
|
|
from voice_changer.utils.VoiceChangerParams import VoiceChangerParams
|
|
|
|
|
|
class PitchExtractorManager(Protocol):
|
|
currentPitchExtractor: PitchExtractor | None = None
|
|
params: VoiceChangerParams
|
|
|
|
@classmethod
|
|
def initialize(cls, params: VoiceChangerParams):
|
|
cls.params = params
|
|
|
|
@classmethod
|
|
def getPitchExtractor(
|
|
cls, pitchExtractorType: PitchExtractorType, gpu: int
|
|
) -> PitchExtractor:
|
|
cls.currentPitchExtractor = cls.loadPitchExtractor(pitchExtractorType, gpu)
|
|
return cls.currentPitchExtractor
|
|
|
|
@classmethod
|
|
def loadPitchExtractor(
|
|
cls, pitchExtractorType: PitchExtractorType, gpu: int
|
|
) -> PitchExtractor:
|
|
if pitchExtractorType == "harvest":
|
|
return HarvestPitchExtractor()
|
|
elif pitchExtractorType == "dio":
|
|
return DioPitchExtractor()
|
|
elif pitchExtractorType == "crepe":
|
|
return CrepePitchExtractor()
|
|
elif pitchExtractorType == "crepe_tiny":
|
|
return CrepeOnnxPitchExtractor(pitchExtractorType, cls.params.crepe_onnx_tiny, gpu)
|
|
elif pitchExtractorType == "crepe_full":
|
|
return CrepeOnnxPitchExtractor(pitchExtractorType, cls.params.crepe_onnx_full, gpu)
|
|
else:
|
|
# return hubert as default
|
|
raise RuntimeError(
|
|
"[Voice Changer] PitchExctractor not found", pitchExtractorType
|
|
)
|