You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.5 KiB
50 lines
1.5 KiB
#!/bin/bash
|
|
|
|
stage=1
|
|
|
|
TARGET_DIR=${MAIN_ROOT}/dataset/ami
|
|
data_folder=${TARGET_DIR}/amicorpus #e.g., /path/to/amicorpus/
|
|
manual_annot_folder=${TARGET_DIR}/ami_public_manual_1.6.2 #e.g., /path/to/ami_public_manual_1.6.2/
|
|
|
|
save_folder=${MAIN_ROOT}/examples/ami/sd0/data
|
|
ref_rttm_dir=${save_folder}/ref_rttms
|
|
meta_data_dir=${save_folder}/metadata
|
|
|
|
set=L
|
|
|
|
. ${MAIN_ROOT}/utils/parse_options.sh || exit 1;
|
|
set -u
|
|
set -o pipefail
|
|
|
|
mkdir -p ${save_folder}
|
|
|
|
if [ ${stage} -le 0 ]; then
|
|
# Download AMI corpus, You need around 10GB of free space to get whole data
|
|
# The signals are too large to package in this way,
|
|
# so you need to use the chooser to indicate which ones you wish to download
|
|
echo "Please follow https://groups.inf.ed.ac.uk/ami/download/ to download the data."
|
|
echo "Annotations: AMI manual annotations v1.6.2 "
|
|
echo "Signals: "
|
|
echo "1) Select one or more AMI meetings: the IDs please follow ./ami_split.py"
|
|
echo "2) Select media streams: Just select Headset mix"
|
|
exit 0;
|
|
fi
|
|
|
|
if [ ${stage} -le 1 ]; then
|
|
echo "AMI Data preparation"
|
|
|
|
python local/ami_prepare.py --data_folder ${data_folder} \
|
|
--manual_annot_folder ${manual_annot_folder} \
|
|
--save_folder ${save_folder} --ref_rttm_dir ${ref_rttm_dir} \
|
|
--meta_data_dir ${meta_data_dir}
|
|
|
|
if [ $? -ne 0 ]; then
|
|
echo "Prepare AMI failed. Please check log message."
|
|
exit 1
|
|
fi
|
|
|
|
fi
|
|
|
|
echo "AMI data preparation done."
|
|
exit 0
|