{ "cells": [ { "cell_type": "code", "execution_count": 1, "id": "9f23e1f8", "metadata": { "ExecuteTime": { "end_time": "2024-01-23T04:15:41.636875Z", "start_time": "2024-01-23T04:15:41.537506Z" } }, "outputs": [], "source": [ "import os\n", "\n", "os.environ[\"CUDA_VISIBLE_DEVICES\"] = \"2\"\n", "\n", "import numpy as np" ] }, { "cell_type": "code", "execution_count": 9, "id": "6dc4f6e7", "metadata": { "ExecuteTime": { "end_time": "2024-01-23T04:17:26.507689Z", "start_time": "2024-01-23T04:17:26.504735Z" } }, "outputs": [], "source": [ "from suno_utils.tasks.dac_2c import (\n", " preload_models as preload_codec_models_8,\n", " decode as codec_decode_8,\n", ")\n", "from suno_utils.tasks.mert_25 import (\n", " preload_models as preload_semantic_models,\n", " encode as semantic_encode,\n", ")\n", "from suno_utils.tasks.dac_2c_12cb import (\n", " preload_models as preload_codec_models_12,\n", " encode as codec_encode_12,\n", " decode as codec_decode_12\n", ")\n", "import os\n", "import numpy as np\n", "from tqdm import tqdm\n", "\n", "\n", "codec_ckpt_path_8 = \"/home/victor/data/models/chirp_v2/dac_2c_25x8.pt\"\n", "semantic_ckpt_path = \"/home/victor/data/models/chirp_v2/mert_25.pt\"\n", "semantic_centroids_path = \"/home/victor/data/models/chirp_v2/mert_25_2x4k.npy\"\n", "codec_ckpt_path_12 = \"/app/suno/tony/v3/dac_2c_25x12.pt\"" ] }, { "cell_type": "code", "execution_count": 4, "id": "620aef25", "metadata": { "ExecuteTime": { "end_time": "2024-01-23T04:16:13.127239Z", "start_time": "2024-01-23T04:16:02.473751Z" } }, "outputs": [], "source": [ "preload_codec_models_8(codec_ckpt_path_8, device=\"cuda\")\n", "preload_codec_models_12(codec_ckpt_path_12, device=\"cuda\")\n", "preload_semantic_models(\n", " semantic_ckpt_path, semantic_centroids_path, device=\"cuda\"\n", ")" ] }, { "cell_type": "code", "execution_count": 7, "id": "3727d346", "metadata": { "ExecuteTime": { "end_time": "2024-01-23T04:17:00.900485Z", "start_time": "2024-01-23T04:17:00.898221Z" } }, "outputs": [], "source": [ "total_jobs = [\n", " \"fff507cb-b970-4583-8cc3-d937dbf0e4b9.npz\",\n", " \"fff82f8b-af8c-484e-a4e7-3bf38ef4c410.npz\",\n", "]\n", "input_dir = \"/app/suno/data/dpo/npz\"\n", "output_dir = \"/app/suno/data/dpo/7b_npz\"" ] }, { "cell_type": "code", "execution_count": 8, "id": "3c514381", "metadata": { "ExecuteTime": { "end_time": "2024-01-23T04:17:01.992205Z", "start_time": "2024-01-23T04:17:01.990127Z" } }, "outputs": [], "source": [ "filename = total_jobs[0]\n", "input_path = os.path.join(input_dir, filename)\n", "output_path = os.path.join(output_dir, filename)" ] }, { "cell_type": "code", "execution_count": 18, "id": "f3feb837", "metadata": { "ExecuteTime": { "end_time": "2024-01-23T04:28:17.405975Z", "start_time": "2024-01-23T04:28:14.138027Z" } }, "outputs": [], "source": [ "npz = np.load(input_path).get(\"v1_raw\")\n", "codec_labels = npz[:, 1:]\n", "audio = codec_decode_8(codec_labels)\n", "semantic_labels = semantic_encode(audio, n_codebooks=1)\n", "codec_labels = codec_encode_12(audio)\n", "n_frames = min(semantic_labels.shape[0], codec_labels.shape[0])\n", "audio_arr = np.concatenate(\n", " [semantic_labels[:n_frames, :], codec_labels[:n_frames, :]],\n", " axis=-1,\n", ")\n", "output_npz = {\"v2_raw\": audio_arr}\n", "audio_12 = codec_decode_12(codec_labels[:n_frames, :])" ] }, { "cell_type": "code", "execution_count": 26, "id": "c05ce27a", "metadata": { "ExecuteTime": { "end_time": "2024-01-23T04:33:31.856524Z", "start_time": "2024-01-23T04:33:31.854257Z" } }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "(1999, 13)\n" ] } ], "source": [ "print(audio_arr.shape)" ] }, { "cell_type": "code", "execution_count": 19, "id": "b8918188", "metadata": { "ExecuteTime": { "end_time": "2024-01-23T04:28:19.696559Z", "start_time": "2024-01-23T04:28:19.694419Z" } }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "80.0 79.96\n" ] } ], "source": [ "print(audio.duration_s, audio_12.duration_s)" ] }, { "cell_type": "code", "execution_count": 24, "id": "9f73f814", "metadata": { "ExecuteTime": { "end_time": "2024-01-23T04:29:19.852280Z", "start_time": "2024-01-23T04:29:19.849915Z" } }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "1999 2000\n" ] } ], "source": [ "print(len(semantic_labels), len(npz[:, 0]))" ] }, { "cell_type": "code", "execution_count": 25, "id": "f44a9e27", "metadata": { "ExecuteTime": { "end_time": "2024-01-23T04:29:53.580152Z", "start_time": "2024-01-23T04:29:53.550269Z" } }, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "0 [3755] 2484\n", "4 [1913] 1034\n", "5 [1579] 3345\n", "10 [109] 2430\n", "12 [109] 3247\n", "14 [3365] 2046\n", "16 [3750] 895\n", "17 [895] 3569\n", "19 [835] 2756\n", "23 [1642] 914\n", "24 [1642] 857\n", "25 [3471] 1642\n", "27 [3604] 2874\n", "28 [1628] 3365\n", "33 [567] 2151\n", "34 [3132] 995\n", "35 [2918] 1222\n", "36 [3637] 3438\n", "37 [3637] 3438\n", "38 [704] 98\n", "40 [1458] 3899\n", "41 [1458] 2205\n", "42 [1458] 2151\n", "43 [1458] 1383\n", "44 [529] 2101\n", "47 [926] 208\n", "49 [2274] 2281\n", "50 [1060] 1170\n", "51 [2410] 1280\n", "52 [1170] 1280\n", "61 [3217] 1280\n", "62 [393] 1394\n", "63 [3518] 542\n", "64 [2460] 3094\n", "65 [177] 1071\n", "73 [3524] 1262\n", "74 [318] 3406\n", "75 [836] 462\n", "78 [2218] 3350\n", "79 [1202] 855\n", "82 [1734] 48\n", "83 [3295] 3391\n", "84 [3293] 1479\n", "87 [2356] 3225\n", "88 [1292] 1000\n", "90 [1556] 2191\n", "92 [2028] 3225\n", "95 [2938] 2145\n", "97 [3909] 143\n", "100 [2948] 2062\n", "103 [966] 313\n", "104 [3609] 3142\n", "106 [3695] 1001\n", "113 [345] 331\n", "115 [3976] 3894\n", "121 [1426] 1491\n", "122 [1279] 3705\n", "123 [80] 3232\n", "129 [985] 2058\n", "133 [2704] 3475\n", "135 [2213] 1108\n", "141 [1880] 652\n", "145 [2137] 1942\n", "149 [2728] 3840\n", "152 [199] 2033\n", "153 [199] 2033\n", "154 [2033] 857\n", "155 [2033] 1949\n", "156 [802] 1949\n", "157 [802] 1949\n", "159 [1949] 1722\n", "160 [1949] 1722\n", "161 [3596] 1722\n", "168 [933] 802\n", "172 [1207] 704\n", "173 [1090] 2101\n", "174 [3366] 2101\n", "175 [1110] 260\n", "176 [2907] 3153\n", "180 [1566] 2221\n", "181 [3637] 1776\n", "182 [3637] 1776\n", "186 [2095] 3945\n", "188 [337] 1201\n", "190 [243] 1201\n", "192 [196] 1201\n", "193 [2460] 1201\n", "199 [1621] 780\n", "201 [2906] 873\n", "202 [1070] 740\n", "203 [1070] 1861\n", "204 [3538] 3826\n", "205 [660] 2452\n", "206 [404] 1936\n", "208 [1133] 1936\n", "209 [1133] 2830\n", "210 [3629] 1785\n", "219 [1732] 286\n", "220 [1951] 286\n", "221 [1951] 286\n", "222 [3289] 286\n", "223 [2207] 2716\n", "225 [3829] 1556\n", "227 [1357] 3468\n", "234 [2956] 1299\n", "237 [272] 313\n", "239 [3577] 1840\n", "251 [3876] 958\n", "252 [528] 396\n", "253 [2580] 1579\n", "256 [1139] 1054\n", "259 [694] 2291\n", "260 [3748] 959\n", "263 [2282] 338\n", "264 [2338] 3764\n", "269 [2227] 3646\n", "275 [1880] 2227\n", "279 [13] 2317\n", "281 [3635] 2548\n", "282 [171] 1949\n", "287 [3373] 3963\n", "293 [809] 3459\n", "295 [845] 2567\n", "296 [2567] 1492\n", "297 [3780] 219\n", "303 [3376] 2883\n", "306 [3053] 2230\n", "310 [3909] 986\n", "312 [2513] 1657\n", "315 [3839] 2468\n", "316 [3503] 1671\n", "320 [3898] 2039\n", "327 [3378] 68\n", "335 [3391] 2938\n", "338 [328] 2466\n", "340 [229] 1652\n", "344 [48] 636\n", "350 [3065] 229\n", "351 [1501] 572\n", "357 [2625] 3219\n", "361 [3681] 1240\n", "367 [3917] 62\n", "372 [345] 301\n", "373 [668] 3577\n", "376 [1777] 599\n", "385 [1605] 2552\n", "386 [2447] 220\n", "387 [3367] 1139\n", "389 [2148] 2206\n", "391 [1223] 1775\n", "395 [3646] 2195\n", "397 [3603] 2548\n", "398 [2157] 723\n", "399 [3840] 2957\n", "402 [2202] 3134\n", "403 [1222] 647\n", "404 [1274] 636\n", "409 [3138] 819\n", "411 [375] 2283\n", "412 [3456] 3298\n", "413 [282] 2065\n", "415 [1941] 375\n", "419 [3123] 3107\n", "421 [3331] 1395\n", "425 [2283] 3935\n", "427 [1418] 74\n", "428 [1835] 4\n", "432 [8] 232\n", "440 [2688] 1731\n", "445 [2144] 234\n", "448 [3804] 1671\n", "451 [1885] 3894\n", "457 [1989] 2062\n", "458 [1422] 150\n", "459 [2716] 1085\n", "465 [1549] 2788\n", "469 [139] 3740\n", "481 [3092] 1878\n", "482 [1862] 143\n", "483 [1322] 2074\n", "485 [737] 3573\n", "486 [3182] 3909\n", "491 [1862] 3968\n", "495 [3607] 2956\n", "497 [80] 211\n", "499 [2819] 80\n", "500 [3385] 2845\n", "509 [3909] 890\n", "510 [1900] 2227\n", "520 [1760] 2957\n", "523 [2461] 1563\n", "528 [2903] 414\n", "531 [2548] 143\n", "534 [1138] 2614\n", "535 [21] 2614\n", "537 [718] 944\n", "538 [944] 3726\n", "549 [2269] 770\n", "551 [11] 2267\n", "553 [3592] 3381\n", "557 [11] 2853\n", "560 [46] 143\n", "562 [1695] 2467\n", "564 [3485] 3912\n", "566 [1077] 8\n", "573 [178] 657\n", "574 [2748] 2245\n", "582 [1612] 524\n", "583 [2688] 1731\n", "586 [2419] 1785\n", "589 [3413] 3740\n", "590 [1698] 1305\n", "593 [41] 2757\n", "594 [1758] 1199\n", "600 [110] 3025\n", "603 [2358] 1453\n", "608 [1545] 3185\n", "611 [1640] 143\n", "617 [1862] 3968\n", "619 [3891] 331\n", "623 [3577] 331\n", "627 [985] 2058\n", "636 [3017] 1572\n", "637 [338] 1572\n", "644 [1449] 2427\n", "645 [652] 1181\n", "647 [1996] 1892\n", "660 [3235] 3127\n", "670 [375] 3803\n", "681 [643] 1360\n", "682 [2269] 1657\n", "683 [2269] 770\n", "684 [3637] 1356\n", "685 [3637] 1776\n", "686 [2039] 3305\n", "690 [74] 303\n", "692 [1794] 3994\n", "695 [3697] 2347\n", "696 [2543] 1657\n", "699 [1730] 625\n", "701 [3391] 3121\n", "704 [1285] 1730\n", "705 [538] 2018\n", "710 [3405] 229\n", "714 [2195] 1652\n", "717 [3695] 723\n", "720 [1210] 2062\n", "721 [559] 1483\n", "725 [2074] 636\n", "726 [814] 737\n", "728 [2608] 737\n", "729 [572] 259\n", "731 [1546] 270\n", "732 [964] 3607\n", "736 [305] 286\n", "738 [339] 286\n", "743 [1840] 301\n", "744 [3879] 1840\n", "745 [1849] 2186\n", "749 [403] 2035\n", "751 [1762] 1465\n", "752 [2211] 1465\n", "753 [2211] 1465\n", "754 [3792] 1505\n", "756 [474] 958\n", "757 [1572] 1236\n", "759 [3328] 2617\n", "762 [1804] 3603\n", "763 [3475] 436\n", "768 [694] 1892\n", "770 [2614] 2957\n", "771 [2634] 1181\n", "773 [175] 218\n", "774 [3074] 625\n", "792 [2920] 3123\n", "801 [3909] 46\n", "806 [3053] 3280\n", "808 [3671] 3685\n", "813 [1885] 1060\n", "815 [1022] 19\n", "816 [2380] 773\n", "817 [1356] 2513\n", "827 [1988] 700\n", "829 [1730] 2245\n", "830 [93] 2910\n", "833 [1104] 18\n", "836 [3797] 993\n", "837 [1785] 636\n", "842 [1208] 1316\n", "843 [41] 3380\n", "851 [763] 1011\n", "852 [2861] 1332\n", "854 [3609] 3142\n", "857 [1527] 3052\n", "859 [1750] 705\n", "868 [1210] 2800\n", "872 [301] 3705\n", "873 [3263] 3232\n", "874 [2504] 1141\n", "875 [1573] 1413\n", "880 [2157] 890\n", "881 [3680] 2166\n", "884 [2157] 1163\n", "887 [338] 1572\n", "889 [3741] 397\n", "890 [474] 2728\n", "891 [895] 3090\n", "892 [199] 1441\n", "893 [2033] 199\n", "894 [1642] 2033\n", "898 [3628] 5\n", "899 [2227] 3897\n", "903 [1628] 2157\n", "904 [1138] 2728\n", "907 [802] 2353\n", "908 [386] 636\n", "909 [3459] 630\n", "911 [3963] 723\n", "919 [2269] 3381\n", "925 [3637] 3756\n", "932 [3147] 2856\n", "933 [8] 670\n", "938 [8] 1167\n", "939 [1776] 1990\n", "948 [1591] 1682\n", "949 [740] 1591\n", "951 [1682] 1617\n", "952 [1682] 993\n", "954 [54] 3887\n", "957 [3429] 1133\n", "962 [2466] 2356\n", "964 [1799] 313\n", "967 [1554] 3818\n", "969 [3408] 1554\n", "970 [1387] 1554\n", "974 [1469] 3804\n", "975 [1157] 356\n", "980 [3065] 737\n", "988 [345] 1297\n", "989 [1297] 814\n", "990 [602] 620\n", "991 [1930] 3393\n", "997 [3887] 3917\n", "999 [1779] 668\n", "1000 [3980] 3969\n", "1005 [1599] 1369\n", "1007 [1210] 1823\n", "1008 [690] 2291\n", "1011 [3680] 1181\n", "1012 [1282] 1223\n", "1016 [1880] 3680\n", "1022 [2353] 3678\n", "1024 [3414] 143\n", "1025 [283] 3127\n", "1026 [283] 3476\n", "1027 [2618] 3476\n", "1029 [1775] 819\n", "1047 [3373] 2065\n", "1050 [1060] 2267\n", "1053 [1657] 884\n", "1054 [287] 2221\n", "1055 [2187] 802\n", "1056 [577] 799\n", "1060 [646] 3298\n", "1065 [1022] 234\n", "1067 [2269] 2513\n", "1075 [1745] 2195\n", "1080 [2906] 1133\n", "1081 [733] 3121\n", "1085 [1436] 538\n", "1086 [1469] 855\n", "1090 [1734] 2779\n", "1100 [1063] 3340\n", "1102 [1208] 634\n", "1104 [1667] 1436\n", "1105 [3274] 636\n", "1107 [1469] 814\n", "1110 [270] 356\n", "1111 [2180] 310\n", "1116 [3856] 1148\n", "1123 [1210] 2800\n", "1125 [3385] 14\n", "1127 [3551] 326\n", "1132 [2364] 1736\n", "1139 [2852] 2479\n", "1141 [1217] 3600\n", "1148 [2118] 2157\n", "1149 [652] 2954\n", "1163 [623] 143\n", "1166 [1492] 3104\n", "1173 [3229] 31\n", "1175 [285] 31\n", "1177 [2920] 2371\n", "1189 [216] 2221\n", "1196 [3016] 771\n", "1197 [2786] 771\n", "1198 [3041] 3534\n", "1199 [3920] 1730\n", "1200 [1799] 343\n", "1201 [3146] 1885\n", "1202 [1212] 855\n", "1204 [1731] 143\n", "1210 [2043] 538\n", "1214 [490] 313\n", "1222 [1765] 2875\n", "1223 [3802] 1229\n", "1224 [3740] 3065\n", "1233 [746] 3553\n", "1238 [211] 2111\n", "1243 [1279] 301\n", "1246 [3129] 3052\n", "1247 [1279] 3705\n", "1255 [2081] 890\n", "1256 [2166] 3680\n", "1260 [2552] 1605\n", "1262 [1823] 3587\n", "1266 [2195] 1181\n", "1269 [330] 2282\n", "1270 [3515] 3060\n", "1274 [1396] 3840\n", "1278 [282] 3515\n", "1282 [2283] 1892\n", "1283 [386] 3646\n", "1285 [7] 3354\n", "1286 [2272] 2317\n", "1287 [926] 1449\n", "1289 [922] 1343\n", "1295 [316] 669\n", "1299 [275] 3894\n", "1300 [1984] 2091\n", "1306 [3303] 71\n", "1309 [3887] 1395\n", "1311 [1060] 1671\n", "1312 [1885] 1060\n", "1313 [1210] 86\n", "1316 [71] 1835\n", "1317 [2690] 636\n", "1320 [3429] 780\n", "1321 [1179] 3121\n", "1322 [3994] 3121\n", "1323 [3529] 1917\n", "1324 [3529] 786\n", "1325 [3529] 1262\n", "1326 [1715] 1262\n", "1327 [1715] 1262\n", "1328 [3947] 1262\n", "1329 [910] 1262\n", "1331 [3947] 1262\n", "1344 [2064] 2786\n", "1349 [3185] 1092\n", "1350 [2] 814\n", "1353 [2074] 143\n", "1354 [1297] 2074\n", "1361 [1240] 3681\n", "1364 [95] 2111\n", "1365 [3577] 2348\n", "1370 [3121] 3573\n", "1371 [3129] 3583\n", "1372 [95] 80\n", "1374 [2819] 80\n", "1375 [730] 636\n", "1376 [131] 1572\n", "1377 [1572] 2282\n", "1378 [1511] 1235\n", "1380 [1728] 982\n", "1388 [694] 2291\n", "1391 [3600] 338\n", "1393 [181] 2353\n", "1394 [890] 3414\n", "1396 [1996] 1823\n", "1401 [1880] 652\n", "1402 [1880] 1641\n", "1407 [904] 2480\n", "1410 [21] 2614\n", "1412 [2272] 1926\n", "1413 [46] 1926\n", "1416 [2272] 3963\n", "1418 [623] 1995\n", "1420 [3456] 646\n", "1421 [446] 2431\n", "1422 [100] 3780\n", "1423 [2269] 1077\n", "1425 [2004] 3637\n", "1426 [3637] 2891\n", "1436 [548] 2692\n", "1438 [3529] 1356\n", "1442 [1621] 3994\n", "1444 [3529] 2861\n", "1452 [1674] 2062\n", "1455 [71] 2593\n", "1456 [1885] 3887\n", "1460 [733] 229\n", "1464 [2043] 2788\n", "1465 [1820] 557\n", "1469 [2788] 2811\n", "1471 [572] 1395\n", "1474 [3609] 3142\n", "1475 [671] 3609\n", "1477 [3695] 1878\n", "1481 [876] 1469\n", "1482 [3393] 2757\n", "1484 [2] 928\n", "1485 [3370] 3318\n", "1492 [1584] 608\n", "1494 [3705] 3897\n", "1495 [3006] 1640\n", "1496 [901] 3612\n", "1497 [2486] 3705\n", "1498 [1047] 805\n", "1501 [1573] 2163\n", "1504 [2639] 1163\n", "1507 [338] 1572\n", "1510 [2859] 2552\n", "1515 [819] 326\n", "1516 [3391] 3724\n", "1518 [3515] 2206\n", "1526 [1775] 3138\n", "1527 [1021] 3354\n", "1535 [275] 2480\n", "1536 [1872] 3331\n", "1542 [46] 386\n", "1551 [383] 3104\n", "1552 [703] 1170\n", "1557 [2380] 1657\n", "1559 [2004] 3220\n", "1560 [2004] 1280\n", "1561 [2004] 1179\n", "1563 [1424] 2976\n", "1565 [1586] 625\n", "1567 [1060] 3121\n", "1574 [1500] 226\n", "1575 [54] 62\n", "1577 [1549] 2419\n", "1578 [660] 3994\n", "1580 [2788] 1734\n", "1584 [41] 2716\n", "1588 [411] 143\n", "1589 [48] 2074\n", "1594 [2593] 71\n", "1595 [3679] 3405\n", "1596 [3065] 3679\n", "1598 [2608] 229\n", "1606 [358] 2470\n", "1613 [3232] 211\n", "1617 [3762] 3025\n", "1618 [572] 1750\n", "1620 [3182] 3583\n", "1621 [3917] 345\n", "1624 [3184] 1779\n", "1629 [3792] 1505\n", "1630 [1091] 2163\n", "1633 [2696] 3101\n", "1635 [1369] 1978\n", "1640 [819] 326\n", "1641 [2926] 1760\n", "1642 [3216] 2310\n", "1644 [3741] 2109\n", "1645 [3545] 852\n", "1646 [895] 3090\n", "1647 [2474] 2033\n", "1651 [2440] 3134\n", "1654 [630] 1775\n", "1655 [630] 3199\n", "1656 [3354] 2548\n", "1658 [3780] 520\n", "1659 [315] 1600\n", "1664 [3138] 3757\n", "1665 [3229] 1562\n", "1666 [2283] 31\n", "1668 [2371] 1009\n", "1674 [809] 2267\n", "1677 [815] 1671\n", "1681 [2347] 990\n", "1685 [986] 3373\n", "1687 [232] 8\n", "1689 [3887] 2138\n", "1693 [646] 2894\n", "1696 [232] 2043\n", "1697 [3641] 8\n", "1699 [3978] 740\n", "1700 [3978] 3826\n", "1703 [3421] 1731\n", "1704 [2748] 580\n", "1705 [1973] 3640\n", "1706 [3378] 544\n", "1709 [3629] 1305\n", "1710 [1456] 3817\n", "1711 [2320] 1483\n", "1712 [2320] 1483\n", "1714 [2195] 2593\n", "1717 [2414] 2608\n", "1718 [1765] 2053\n", "1727 [3102] 2875\n", "1730 [2111] 1063\n", "1734 [1556] 2800\n", "1737 [2336] 1862\n", "1747 [331] 668\n", "1751 [1736] 2166\n", "1752 [1588] 1823\n", "1756 [3603] 3887\n", "1758 [3030] 3969\n", "1760 [1572] 326\n", "1761 [3959] 1139\n", "1765 [3959] 1775\n", "1766 [1228] 1139\n", "1768 [2637] 647\n", "1769 [1880] 2227\n", "1773 [2427] 386\n", "1780 [922] 636\n", "1783 [3798] 638\n", "1784 [3414] 3938\n", "1786 [3963] 2118\n", "1787 [386] 3414\n", "1791 [155] 3280\n", "1792 [520] 208\n", "1793 [3690] 458\n", "1794 [770] 3671\n", "1795 [3671] 2966\n", "1796 [3671] 1995\n", "1798 [863] 3963\n", "1803 [1170] 3381\n", "1804 [670] 1424\n", "1806 [643] 3100\n", "1807 [2269] 1657\n", "1809 [3220] 3592\n", "1810 [3887] 3592\n", "1812 [162] 1870\n", "1814 [986] 723\n", "1815 [2347] 2039\n", "1816 [858] 3391\n", "1817 [2218] 858\n", "1818 [1280] 1179\n", "1819 [1179] 740\n", "1820 [3826] 1179\n", "1821 [544] 1179\n", "1827 [1500] 544\n", "1828 [1500] 544\n", "1832 [1500] 3817\n", "1838 [1645] 1765\n", "1840 [1306] 3817\n", "1841 [2875] 1645\n", "1844 [2875] 1645\n", "1846 [2161] 2875\n", "1847 [2875] 1765\n", "1848 [253] 1765\n", "1851 [2180] 3589\n", "1852 [2757] 3706\n", "1855 [3829] 1556\n", "1856 [3277] 1323\n", "1861 [616] 2971\n", "1862 [616] 746\n", "1863 [2971] 746\n", "1864 [616] 2971\n", "1865 [1398] 2971\n", "1866 [1452] 3047\n", "1867 [3667] 272\n", "1870 [616] 1148\n", "1875 [1686] 1150\n", "1880 [2859] 2471\n", "1884 [3345] 2715\n", "1886 [1654] 2252\n", "1889 [2252] 3495\n", "1892 [2715] 2252\n", "1894 [2252] 2715\n", "1895 [2252] 1654\n", "1897 [2715] 2252\n", "1898 [2506] 1654\n", "1916 [3546] 2506\n", "1917 [1379] 2506\n", "1918 [739] 2506\n", "1919 [481] 2506\n", "1920 [481] 2506\n", "1923 [2506] 481\n", "1924 [3546] 481\n", "1926 [2907] 3172\n", "1929 [1516] 770\n", "1931 [579] 216\n", "1932 [216] 863\n", "1933 [1237] 3016\n", "1936 [863] 3016\n", "1940 [234] 3552\n", "1944 [480] 216\n", "1945 [480] 1072\n", "1948 [1072] 3534\n", "1949 [1529] 1072\n", "1950 [1529] 2238\n", "1954 [1183] 3552\n", "1960 [3389] 3817\n", "1963 [2875] 1765\n", "1965 [2875] 1765\n", "1967 [2875] 1765\n", "1970 [2875] 1765\n", "1979 [2161] 1765\n", "1980 [2161] 1765\n", "1981 [2161] 2971\n", "1982 [238] 1765\n", "1983 [238] 746\n", "1984 [238] 2971\n", "1985 [238] 2161\n", "1986 [1650] 238\n", "1989 [1650] 238\n", "1996 [2330] 616\n", "1998 [403] 2330\n" ] } ], "source": [ "for i, (new_s, old_s) in enumerate(zip(semantic_labels, npz[:, 0])):\n", " if new_s != old_s:\n", " print(i, new_s, old_s)" ] }, { "cell_type": "code", "execution_count": 11, "id": "d64570b2", "metadata": { "ExecuteTime": { "end_time": "2024-01-23T04:17:53.363587Z", "start_time": "2024-01-23T04:17:52.078341Z" } }, "outputs": [ { "data": { "text/html": [ "\n", "\n" ], "text/plain": [ "" ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "audio.play()" ] }, { "cell_type": "code", "execution_count": 12, "id": "d32e8ed5", "metadata": { "ExecuteTime": { "end_time": "2024-01-23T04:17:58.636240Z", "start_time": "2024-01-23T04:17:57.321889Z" } }, "outputs": [ { "data": { "text/html": [ "\n", "\n" ], "text/plain": [ "" ] }, "metadata": {}, "output_type": "display_data" } ], "source": [ "audio_12.play()" ] }, { "cell_type": "code", "execution_count": 28, "id": "2e69b929", "metadata": { "ExecuteTime": { "end_time": "2024-01-23T04:36:44.113042Z", "start_time": "2024-01-23T04:36:44.110771Z" } }, "outputs": [], "source": [ "test_load = np.load(output_path)" ] }, { "cell_type": "code", "execution_count": 31, "id": "abb7c961", "metadata": { "ExecuteTime": { "end_time": "2024-01-23T04:36:56.505557Z", "start_time": "2024-01-23T04:36:56.502311Z" } }, "outputs": [ { "data": { "text/plain": [ "(1999, 13)" ] }, "execution_count": 31, "metadata": {}, "output_type": "execute_result" } ], "source": [ "test_load[\"v2_raw\"].shape" ] }, { "cell_type": "code", "execution_count": null, "id": "b1475394", "metadata": {}, "outputs": [], "source": [] } ], "metadata": { "kernelspec": { "display_name": "Python 3 (ipykernel)", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.10.12" }, "toc": { "base_numbering": 1, "nav_menu": {}, "number_sections": true, "sideBar": true, "skip_h1_title": false, "title_cell": "Table of Contents", "title_sidebar": "Contents", "toc_cell": false, "toc_position": {}, "toc_section_display": true, "toc_window_display": false } }, "nbformat": 4, "nbformat_minor": 5 }