add patch from Damien.Allain
This commit is contained in:
parent
fe4129a4ee
commit
f5d4a76db9
@ -5,18 +5,25 @@ import sys
|
|||||||
import os
|
import os
|
||||||
import wave
|
import wave
|
||||||
|
|
||||||
|
modelDir=os.path.dirname(sys.argv[0])+"/models/fr"
|
||||||
|
|
||||||
SetLogLevel(0)
|
SetLogLevel(0)
|
||||||
if not os.path.exists("models/fr"):
|
if not os.path.exists(modelDir):
|
||||||
print ("Please download the model from https://alphacephei.com/vosk/models and unpack as 'models' in the current folder.")
|
print ("Please download the model from https://alphacephei.com/vosk/models and unpack as '{}'.".format(modelDir))
|
||||||
exit (1)
|
exit (1)
|
||||||
|
|
||||||
|
|
||||||
wf = wave.open(sys.argv[1], "rb")
|
wf = wave.open(sys.argv[1], "rb")
|
||||||
if wf.getnchannels() != 1 or wf.getsampwidth() != 2 or wf.getcomptype() != "NONE":
|
framerate=wf.getframerate()
|
||||||
print ("Audio file must be WAV format mono PCM.")
|
nchannels=wf.getnchannels()
|
||||||
|
sampwidth=wf.getsampwidth()
|
||||||
|
comptype=wf.getcomptype()
|
||||||
|
if framerate < 16000 or nchannels != 1 or sampwidth != 2 or comptype != "NONE":
|
||||||
|
print ("Audio file has : {} Hz sample rate, {} channels, {} byte sample width and {} compression type".format(framerate, nchannels, sampwidth, comptype))
|
||||||
|
print ("when 16000 Hz sample rate, 1 channel, 2 byte sample width and NONE compression type are required")
|
||||||
exit (1)
|
exit (1)
|
||||||
|
|
||||||
model = Model("models/fr")
|
model = Model(modelDir)
|
||||||
rec = KaldiRecognizer(model, wf.getframerate())
|
rec = KaldiRecognizer(model, wf.getframerate())
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
|
17
install.sh
17
install.sh
@ -26,23 +26,24 @@ echo -e " ${reset}"
|
|||||||
pip3 -v install vosk
|
pip3 -v install vosk
|
||||||
echo "${green}########### récupération du modèle de reconnaissance en Français sous licence aGPL, taille: 1.6Go. Choix des modèles disponibles: https://alphacephei.com/vosk/models ${reset}"
|
echo "${green}########### récupération du modèle de reconnaissance en Français sous licence aGPL, taille: 1.6Go. Choix des modèles disponibles: https://alphacephei.com/vosk/models ${reset}"
|
||||||
echo " "
|
echo " "
|
||||||
mkdir -p models/fr
|
mkdir -p models
|
||||||
echo -e "${green}########### Procéder au téléchargement du modèle Français (1.6go) pour transcrire les textes ?${reset} (écrivez o pour oui et faites entrée pour valider) ${reset}"
|
echo -e "${green}########### Procéder au téléchargement du modèle Français (1.6go) pour transcrire les textes ?${reset} (écrivez o pour oui et faites entrée pour valider) ${reset}"
|
||||||
read proceed
|
read proceed
|
||||||
|
|
||||||
# les autres modèles sont ici https://alphacephei.com/vosk/models
|
# les autres modèles sont ici https://alphacephei.com/vosk/models
|
||||||
if [[ $proceed == o* ]]; then
|
if [[ $proceed == o* ]]; then
|
||||||
echo "C'est parti."
|
echo "C'est parti."
|
||||||
if [ ! -f "vosk-model-fr-0.6-linto-2.2.0.zip" ]; then
|
f="vosk-model-fr-0.6-linto-2.2.0.zip"
|
||||||
wget https://alphacephei.com/vosk/models/vosk-model-fr-0.6-linto-2.2.0.zip
|
if [ "x`find $f -printf %s`" != x1582884866 ]; then
|
||||||
|
wget https://alphacephei.com/vosk/models/$f
|
||||||
echo -e "${green}########### téléchargement du modèle OK ${reset}"
|
echo -e "${green}########### téléchargement du modèle OK ${reset}"
|
||||||
unzip vosk-model-fr-0.6-linto-2.2.0.zip
|
|
||||||
mv vosk-model-fr-0.6-linto-2.2.0 models/fr
|
|
||||||
echo -e "${green}########### décompression du modèle en Français OK ${reset}"
|
|
||||||
ls -l models/fr
|
|
||||||
else
|
else
|
||||||
echo "fichier zip vosk-model-fr-0.6-linto-2.2.0.zip déjà présent"
|
echo "fichier zip $f déjà présent"
|
||||||
fi
|
fi
|
||||||
|
unzip $f
|
||||||
|
mv vosk-model-fr-0.6-linto-2.2.0 models/fr
|
||||||
|
echo -e "${green}########### décompression du modèle en Français OK ${reset}"
|
||||||
|
ls -l models/fr
|
||||||
else
|
else
|
||||||
echo -e "${green}########### vous n'avez pas souhaité télécharger le modèle ${reset}"
|
echo -e "${green}########### vous n'avez pas souhaité télécharger le modèle ${reset}"
|
||||||
echo -e "${green}########### fin de l'installation sans télécharger de modèle de langue ${reset}"
|
echo -e "${green}########### fin de l'installation sans télécharger de modèle de langue ${reset}"
|
||||||
|
@ -77,7 +77,7 @@ else
|
|||||||
fi
|
fi
|
||||||
echo " "
|
echo " "
|
||||||
|
|
||||||
mkdir output/$FILE_NAME -p
|
mkdir -p "$OUT_DIR"
|
||||||
echo " convertir en sous titre ? $ENABLE_SRT"
|
echo " convertir en sous titre ? $ENABLE_SRT"
|
||||||
if ($ENABLE_SRT) ; then
|
if ($ENABLE_SRT) ; then
|
||||||
echo "########### $(date) : conversion de $file ,sortie en fichier de sous titres .srt"
|
echo "########### $(date) : conversion de $file ,sortie en fichier de sous titres .srt"
|
||||||
|
Loading…
Reference in New Issue
Block a user