Merge pull request #10 from nttcslab/2024t2/v3.0.0

Updated for DCASE2024 Task2

01_train_2024t2.sh

@@ -0,0 +1,40 @@
+TZ=JST-9 date
+
+echo $0 $*
+
+dev_eval=$1
+echo -e "\tdev_eval = '$dev_eval'"
+echo
+
+# check args
+if [ "${dev_eval}" != "-d" ] \
+    && [ "${dev_eval}" != "-e" ] \
+    && [ "${dev_eval}" != "--dev" ] \
+    && [ "${dev_eval}" != "--eval" ]
+then
+    echo "$0: argument error"
+    echo -e "usage\t: $0 ['-d' | '--dev' | '-e' | '--eval']"
+    echo -e "\tinvalid choice '$dev_eval'"
+    echo -e "\tchoice from ['-d' | '--dev' | '-e' | '--eval']."
+    echo -e "\t\t-d, --dev\t: Using Development dataset. "
+    echo -e "\t\t-e, --eval\t: Using Additional training dataset and Evaluation dataset. "
+    echo -e "example\t: $ bash $0 -d"
+    exit 1
+fi
+
+# main process
+base_job="bash"
+job="train_ae.sh"
+
+if [ "${dev_eval}" = "-d" ] || [ "${dev_eval}" = "--dev" ]
+then
+    dataset_list="DCASE2024T2bearing DCASE2024T2fan DCASE2024T2gearbox DCASE2024T2slider DCASE2024T2ToyCar DCASE2024T2ToyTrain DCASE2024T2valve"
+elif [ "${dev_eval}" = "-e" ] || [ "${dev_eval}" = "--eval" ]
+then
+    echo eval data has not been published yet.
+    exit 1
+fi
+
+for dataset in $dataset_list; do
+    ${base_job} ${job} ${dataset} ${dev_eval} 0
+done

02a_test_2024t2.sh

@@ -0,0 +1,40 @@
+TZ=JST-9 date
+
+echo $0 $*
+
+dev_eval=$1
+echo -e "\tdev_eval = '$dev_eval'"
+echo
+
+# check args
+if [ "${dev_eval}" != "-d" ] \
+    && [ "${dev_eval}" != "-e" ] \
+    && [ "${dev_eval}" != "--dev" ] \
+    && [ "${dev_eval}" != "--eval" ]
+then
+    echo "$0: argument error"
+    echo -e "usage\t: $0 ['-d' | '--dev' | '-e' | '--eval']"
+    echo -e "\tinvalid choice '$dev_eval'"
+    echo -e "\tchoice from ['-d' | '--dev' | '-e' | '--eval']."
+    echo -e "\t\t-d, --dev\t: Using Development dataset. "
+    echo -e "\t\t-e, --eval\t: Using Additional training dataset and Evaluation dataset. "
+    echo -e "example\t: $ bash $0 -d"
+    exit 1
+fi
+
+# main process
+base_job="bash"
+job="test_ae.sh"
+
+if [ "${dev_eval}" = "-d" ] || [ "${dev_eval}" = "--dev" ]
+then
+    dataset_list="DCASE2024T2bearing DCASE2024T2fan DCASE2024T2gearbox DCASE2024T2slider DCASE2024T2ToyCar DCASE2024T2ToyTrain DCASE2024T2valve"
+elif [ "${dev_eval}" = "-e" ] || [ "${dev_eval}" = "--eval" ]
+then
+    echo eval data has not been published yet.
+    exit 1
+fi
+
+for dataset in $dataset_list; do
+    ${base_job} ${job} ${dataset} ${dev_eval} "MSE" 0
+done

02b_test_2024t2.sh

@@ -0,0 +1,40 @@
+TZ=JST-9 date
+
+echo $0 $*
+
+dev_eval=$1
+echo -e "\tdev_eval = '$dev_eval'"
+echo
+
+# check args
+if [ "${dev_eval}" != "-d" ] \
+    && [ "${dev_eval}" != "-e" ] \
+    && [ "${dev_eval}" != "--dev" ] \
+    && [ "${dev_eval}" != "--eval" ]
+then
+    echo "$0: argument error"
+    echo -e "usage\t: $0 ['-d' | '--dev' | '-e' | '--eval']"
+    echo -e "\tinvalid choice '$dev_eval'"
+    echo -e "\tchoice from ['-d' | '--dev' | '-e' | '--eval']."
+    echo -e "\t\t-d, --dev\t: Using Development dataset. "
+    echo -e "\t\t-e, --eval\t: Using Additional training dataset and Evaluation dataset. "
+    echo -e "example\t: $ bash $0 -d"
+    exit 1
+fi
+
+# main process
+base_job="bash"
+job="test_ae.sh"
+
+if [ "${dev_eval}" = "-d" ] || [ "${dev_eval}" = "--dev" ]
+then
+    dataset_list="DCASE2024T2bearing DCASE2024T2fan DCASE2024T2gearbox DCASE2024T2slider DCASE2024T2ToyCar DCASE2024T2ToyTrain DCASE2024T2valve"
+elif [ "${dev_eval}" = "-e" ] || [ "${dev_eval}" = "--eval" ]
+then
+    echo eval data has not been published yet.
+    exit 1
+fi
+
+for dataset in $dataset_list; do
+    ${base_job} ${job} ${dataset} ${dev_eval} "MAHALA" 0
+done

03_summarize_results.sh

@@ -14,7 +14,8 @@ args_flag_dataset=0
 if [ "${dataset}" != "DCASE2020T2" ] \
     && [ "${dataset}" != "DCASE2021T2" ] \
     && [ "${dataset}" != "DCASE2022T2" ] \
-    && [ "${dataset}" != "DCASE2023T2" ]
+    && [ "${dataset}" != "DCASE2023T2" ] \
+    && [ "${dataset}" != "DCASE2024T2" ]
 then
     args_flag=1
     args_flag_dataset=1
@@ -33,16 +34,17 @@ fi
 if [ $args_flag -eq 1 ]
 then
     echo "$0: argument error"
-    echo -e "usage\t: $0 ['DCASE2020T2' | 'DCASE2021T2' | 'DCASE2022T2' | 'DCASE2023T2' ] ['-d' | '--dev' | '-e' | '--eval']"
+    echo -e "usage\t: $0 ['DCASE2020T2' | 'DCASE2021T2' | 'DCASE2022T2' | 'DCASE2023T2' | 'DCASE2024T2' ] ['-d' | '--dev' | '-e' | '--eval']"
 
     if [ $args_flag_dataset -eq 1 ]
     then
         echo -e "\tdataset: invalid choice '$dataset'"
-        echo -e "\tchoice from ['DCASE2020T2' | 'DCASE2021T2' | 'DCASE2022T2' | 'DCASE2023T2' ]."
+        echo -e "\tchoice from ['DCASE2020T2' | 'DCASE2021T2' | 'DCASE2022T2' | 'DCASE2023T2' | 'DCASE2024T2' ]."
         echo -e "\t\tDCASE2020T2\t: Use DCASE2020 Task2 datasets. "
         echo -e "\t\tDCASE2021T2\t: Use DCASE2021 Task2 datasets. "
         echo -e "\t\tDCASE2022T2\t: Use DCASE2022 Task2 datasets. "
         echo -e "\t\tDCASE2023T2\t: Use DCASE2023 Task2 datasets. "
+        echo -e "\t\tDCASE2023T2\t: Use DCASE2024 Task2 datasets. "
         echo 
     fi