Upload Audio_Effects_SDK/samples/effects_delayed_streams_demo/run_effect_chained.sh with huggingface_hub

Audio_Effects_SDK/samples/effects_delayed_streams_demo/run_effect_chained.sh

@@ -0,0 +1,152 @@
+#!/bin/bash
+
+# Config file location
+cfg_file=/tmp/tmp_cfg.txt
+
+effect1="denoiser"
+effect2="superres"
+gpu="t4"
+ulimit_val=0
+sample_rate1=16
+sample_rate2=16
+output_sample_rate1=16
+output_sample_rate2=48
+batch_size=10
+reset=
+multi_input=""
+frame_size=10;
+enable_vad=
+ALL_STREAMS_ACTIVE="one_step_delay_streams none
+two_step_delay_streams none"
+SOME_STREAMS_ACTIVE="one_step_delay_streams 0 2 6 8
+two_step_delay_streams 3 5 9"
+
+active_streams=$SOME_STREAMS_ACTIVE
+
+declare -A gpu_map=( ["a100"]="sm_80"
+                     ["a30"]="sm_80"
+                     ["a2"]="sm_86"
+                     ["a10"]="sm_86"
+                     ["a16"]="sm_86"
+                     ["a40"]="sm_86"
+                     ["t4"]="sm_75"
+                     ["v100"]="sm_70"
+                    )
+declare -a supported_effects=( "denoiser_16k_superres_16k"
+                               "dereverb_16k_superres_16k"
+                               "dereverb_denoiser_16k_superres_16k"
+                               "superres_8k_denoiser_16k"
+                               "superres_8k_dereverb_16k"
+                               "superres_8k_dereverb_denoiser_16k"
+                               "superres_8k_dereverb_denoiser_16k"
+                             )
+
+gpu_list=$(IFS=/; printf "%s" "${!gpu_map[*]}")
+
+while [[ "$#" -gt 0 ]]; do
+    case $1 in
+        -e1|--effect1) effect1="$2"; shift ;;
+        -e2|--effect2) effect2="$2"; shift ;;
+        -g|--gpu) gpu="$2"; shift ;;
+        -u|--ulimit) ulimit_val="$2"; shift ;;
+        -s1|--sample_rate) sample_rate1="$2"; shift ;;
+        -s2|--sample_rate2) sample_rate2="$2"; shift ;;
+        -o1|--output_sample_rate1) output_sample_rate1="$2"; shift ;;
+        -o2|--output_sample_rate2) output_sample_rate2="$2"; shift ;;
+        -c|--cfg-file) cfg_file="$2"; shift;;
+        -r|--reset) multi_input="0,1,2,3,4" reset="reset 2 3 5";;
+        -f|--frame_size) frame_size="$2"; shift;;
+        -v|--enable_vad) enable_vad=1;;
+        -t|--all_streams_active) all_streams_active=$ALL_STREAMS_ACTIVE;;
+        -h|--help) echo -e "-g, --gpu : $gpu_list [default=t4]"
+                   echo -e "-e1, --effect1 : First effect to use: denoiser/dereverb/dereverb_denoiser/aec/superres [default=denoiser]"
+                   echo -e "-e2, --effect2 : Second effect to use: denoiser/dereverb/dereverb_denoiser/aec/superres [default=denoiser]"
+                   echo -e "-u, --ulimit : raise files open ulimit, this is used when running larger batch sizes that exceed the kernel maximum open files limit. Warning: Use this option with caution. [default=0(unusued)]"
+                   echo -e "-s1, --sample_rate1 : sample rate to use 8/16/48 (in kHz) for first effect. [default=16]"
+                   echo -e "-s2, --sample_rate2 : sample rate to use 8/16/48 (in kHz) for second effect. [default=48]"
+                   echo -e "-o1, --output_sample_rate1 : output sample rate to use 16/48 (in kHz) for first effect. [default=48]"
+                   echo -e "-o2, --output_sample_rate2 : output sample rate to use 16/48 (in kHz) for second effect. [default=48]"
+                   echo -e "-c, --cfg-file: configuration file to write to"
+                   echo -e "-f, --frame_size: frame size to be used 10/20ms. [default=10]"
+                   echo -e "-v, --enable_vad: Enable VAD. Refer to the Programming Guide for supported effects. [default=disabled]"
+                   echo -e "-t, --all_streams_active: Make all streams active (streams are delayed by default)"
+                   echo -e "-h, --help: print this help menu";exit;;
+        *) echo "Unknown parameter passed: $1"; exit 1 ;;
+    esac
+    shift
+done
+
+effect_check="\b${effect1}_${sample_rate1}k_${effect2}_${sample_rate2}k\b"
+if ! [[ "${supported_effects[@]}" =~ ${effect_check} ]]; then
+  echo "Chain (${effect1} ${sample_rate1}k + ${effect2} ${sample_rate2}k) is not supported. Supported chains are:"
+  for e in "${supported_effects[@]}"; do
+    v=${e//k_/k + }
+    echo "- ${v//_/ }"
+  done
+  exit 1
+fi
+input_list=""
+folder="${effect1}_${sample_rate1}k_${effect2}_${sample_rate2}k"
+wav=$(./list.sh ../input_files/chaining/${folder}/ ${batch_size} ${multi_input})
+if [ $? -ne 0 ]; then
+   echo "Error generating input files list"
+   echo "Command returned: $wav"
+   exit 1
+fi
+input_list="input_wav_list $wav"
+
+out_file=""
+for i in $(seq 1 $batch_size); do
+    out_file="$out_file ${effect1}${sample_rate1}k_${effect2}${sample_rate2}k_$i.wav"
+done
+
+if [ -z "${gpu_map[$gpu]:-}" ]; then
+  echo "Invalid GPU (got $gpu, supported $gpu_list)"
+  exit 1
+fi
+arch="${gpu_map[$gpu]}"
+
+model1="../../models"
+model2="../../models"
+if [ "$effect1" == "superres" ]; then
+  model1="${model1}/${arch}/${effect1}_${sample_rate1}k_to_${output_sample_rate1}k.trtpkg"
+else
+  model1="${model1}/${arch}/${effect1}_${sample_rate1}k.trtpkg"
+fi
+
+if [ "$effect2" == "superres" ]; then
+  model2="${model2}/${arch}/${effect2}_${sample_rate2}k_to_${output_sample_rate2}k.trtpkg"
+else
+  model2="${model2}/${arch}/${effect2}_${sample_rate2}k.trtpkg"
+fi
+
+echo "effect ${effect1} ${effect2}
+sample_rate ${sample_rate1}000 ${sample_rate2}000
+model ${model1} ${model2}
+real_time 0
+$active_streams
+$input_list
+output_wav_list $out_file
+frame_size ${frame_size}" > $cfg_file
+
+if [ ! -z "$reset" ]; then
+  echo "$reset" >> $cfg_file
+fi
+
+if [ ! -z "$enable_vad" ]; then
+  echo "enable_vad 1" >> $cfg_file
+fi
+
+echo "Using the config file $cfg_file"
+
+if [ "$ulimit_val" -ne "0" ]; then
+   if [ "$ulimit_val" -le "1024" ]; then
+      echo "You probably dont want to do this. ulimit is too low.";
+      exit;
+   fi
+   echo "ulimit set to ${ulimit_val}";
+   ulimit -n $ulimit_val
+   exit;
+fi
+
+./effects_delayed_streams_demo -c $cfg_file