This commit is contained in:
Jakub Adamski 2023-06-14 10:59:08 +02:00
parent 1a8df0cf88
commit 3bcce61a46
3 changed files with 343 additions and 33 deletions

4
.gitignore vendored
View File

@ -1 +1,3 @@
.conda/ .conda/
output/
*.mp4

291
SE8.ipynb

File diff suppressed because one or more lines are too long

81
data.csv Normal file
View File

@ -0,0 +1,81 @@
angry0,box0,disgust0,fear0,happy0,neutral0,sad0,surprise0
0.19,"[72, 192, 207, 269]",0.0,0.06,0.0,0.58,0.17,0.0
0.14,"[72, 192, 208, 269]",0.0,0.04,0.0,0.68,0.14,0.0
0.2,"[71, 191, 208, 271]",0.0,0.02,0.0,0.62,0.15,0.0
0.24,"[71, 192, 208, 267]",0.0,0.02,0.0,0.59,0.15,0.0
0.28,"[72, 193, 206, 266]",0.0,0.02,0.0,0.56,0.13,0.0
0.33,"[69, 189, 209, 265]",0.0,0.04,0.0,0.51,0.11,0.0
0.3,"[70, 191, 205, 262]",0.0,0.02,0.0,0.56,0.12,0.0
0.25,"[69, 190, 206, 263]",0.0,0.02,0.0,0.62,0.11,0.0
0.22,"[69, 190, 206, 262]",0.0,0.02,0.0,0.62,0.14,0.0
0.17,"[73, 189, 203, 264]",0.0,0.02,0.0,0.67,0.14,0.0
0.17,"[72, 188, 203, 262]",0.0,0.02,0.0,0.7,0.1,0.0
0.16,"[72, 187, 202, 263]",0.0,0.02,0.0,0.7,0.11,0.0
0.12,"[72, 187, 204, 263]",0.0,0.02,0.0,0.78,0.08,0.0
0.1,"[72, 186, 204, 263]",0.0,0.03,0.0,0.79,0.09,0.0
0.15,"[70, 185, 206, 264]",0.0,0.03,0.0,0.7,0.12,0.0
0.18,"[69, 186, 206, 260]",0.0,0.04,0.0,0.64,0.14,0.0
0.14,"[70, 188, 205, 255]",0.0,0.04,0.0,0.67,0.14,0.0
0.17,"[71, 189, 204, 254]",0.0,0.03,0.0,0.63,0.17,0.0
0.36,"[61, 180, 213, 265]",0.0,0.01,0.0,0.54,0.08,0.0
0.24,"[71, 189, 203, 257]",0.0,0.03,0.0,0.59,0.14,0.0
0.15,"[68, 188, 204, 258]",0.0,0.03,0.0,0.64,0.17,0.0
0.24,"[64, 181, 210, 265]",0.0,0.03,0.0,0.49,0.24,0.0
0.35,"[64, 179, 209, 264]",0.0,0.02,0.0,0.5,0.13,0.0
0.44,"[69, 182, 205, 255]",0.0,0.02,0.0,0.31,0.23,0.0
0.41,"[71, 182, 202, 252]",0.0,0.01,0.0,0.27,0.3,0.0
0.44,"[71, 180, 201, 251]",0.0,0.01,0.0,0.3,0.25,0.0
0.41,"[65, 177, 207, 258]",0.0,0.02,0.0,0.34,0.23,0.0
0.48,"[73, 181, 200, 248]",0.0,0.03,0.0,0.2,0.29,0.0
0.4,"[72, 179, 201, 251]",0.0,0.05,0.0,0.21,0.34,0.0
0.39,"[61, 172, 208, 257]",0.0,0.03,0.0,0.42,0.15,0.0
0.4,"[63, 174, 205, 252]",0.0,0.04,0.0,0.37,0.19,0.0
0.49,"[64, 175, 205, 252]",0.0,0.04,0.0,0.22,0.25,0.0
0.42,"[65, 174, 204, 251]",0.0,0.03,0.0,0.32,0.21,0.0
0.34,"[64, 175, 204, 250]",0.0,0.06,0.0,0.33,0.27,0.0
0.42,"[64, 174, 205, 251]",0.0,0.05,0.0,0.28,0.24,0.0
0.45,"[64, 174, 205, 251]",0.0,0.05,0.0,0.26,0.23,0.0
0.48,"[63, 173, 206, 253]",0.0,0.04,0.0,0.27,0.2,0.01
0.38,"[62, 173, 207, 255]",0.0,0.04,0.0,0.36,0.21,0.01
0.35,"[62, 173, 207, 257]",0.0,0.03,0.0,0.38,0.22,0.01
0.43,"[66, 176, 202, 254]",0.0,0.05,0.01,0.26,0.25,0.0
0.33,"[65, 176, 206, 258]",0.0,0.04,0.0,0.47,0.15,0.0
0.17,"[65, 175, 207, 260]",0.0,0.05,0.01,0.64,0.14,0.0
0.22,"[65, 174, 209, 263]",0.0,0.02,0.01,0.6,0.15,0.0
0.14,"[66, 177, 208, 259]",0.0,0.02,0.07,0.67,0.11,0.0
0.07,"[67, 173, 210, 262]",0.0,0.01,0.35,0.52,0.05,0.0
0.02,"[67, 174, 209, 262]",0.0,0.01,0.66,0.29,0.03,0.0
0.0,"[68, 175, 208, 260]",0.0,0.0,0.89,0.1,0.0,0.0
0.0,"[72, 176, 202, 256]",0.0,0.0,0.93,0.07,0.0,0.0
0.0,"[69, 174, 204, 259]",0.0,0.0,0.97,0.03,0.0,0.0
0.0,"[65, 173, 208, 266]",0.0,0.0,0.99,0.01,0.0,0.0
0.0,"[61, 172, 213, 269]",0.0,0.0,1.0,0.0,0.0,0.0
0.0,"[61, 174, 211, 270]",0.0,0.0,1.0,0.0,0.0,0.0
0.0,"[63, 174, 210, 269]",0.0,0.0,1.0,0.0,0.0,0.0
0.0,"[63, 174, 210, 270]",0.0,0.0,1.0,0.0,0.0,0.0
0.0,"[63, 175, 213, 271]",0.0,0.0,1.0,0.0,0.0,0.0
0.0,"[64, 174, 209, 271]",0.0,0.0,0.99,0.0,0.0,0.0
0.0,"[64, 173, 210, 271]",0.0,0.0,0.99,0.01,0.0,0.0
0.0,"[66, 174, 209, 271]",0.0,0.0,0.99,0.01,0.0,0.0
0.0,"[67, 174, 208, 270]",0.0,0.0,0.99,0.01,0.0,0.0
0.0,"[66, 173, 211, 271]",0.0,0.0,0.99,0.01,0.0,0.0
0.0,"[62, 172, 224, 271]",0.0,0.0,1.0,0.0,0.0,0.0
0.0,"[65, 171, 216, 276]",0.0,0.0,0.99,0.01,0.0,0.0
0.0,"[62, 172, 224, 270]",0.0,0.0,1.0,0.0,0.0,0.0
0.0,"[60, 172, 224, 272]",0.0,0.0,0.99,0.01,0.0,0.0
0.0,"[67, 173, 214, 274]",0.0,0.0,0.99,0.01,0.0,0.0
0.0,"[60, 172, 223, 272]",0.0,0.0,1.0,0.0,0.0,0.0
0.0,"[61, 171, 224, 270]",0.0,0.0,1.0,0.0,0.0,0.0
0.0,"[62, 172, 223, 270]",0.0,0.0,1.0,0.0,0.0,0.0
0.0,"[66, 170, 213, 275]",0.0,0.0,0.99,0.01,0.0,0.0
0.0,"[59, 171, 225, 270]",0.0,0.0,1.0,0.0,0.0,0.0
0.0,"[61, 172, 223, 269]",0.0,0.0,1.0,0.0,0.0,0.0
0.0,"[60, 172, 223, 270]",0.0,0.0,1.0,0.0,0.0,0.0
0.0,"[61, 171, 223, 270]",0.0,0.0,1.0,0.0,0.0,0.0
0.0,"[64, 165, 216, 280]",0.0,0.0,0.99,0.01,0.0,0.0
0.0,"[60, 170, 222, 271]",0.0,0.0,0.99,0.01,0.0,0.0
0.0,"[63, 172, 221, 270]",0.0,0.0,1.0,0.0,0.0,0.0
0.0,"[66, 174, 209, 269]",0.0,0.0,0.98,0.02,0.0,0.0
0.0,"[70, 178, 211, 264]",0.0,0.0,0.93,0.07,0.0,0.0
0.01,"[71, 181, 209, 266]",0.0,0.0,0.58,0.4,0.0,0.0
0.03,"[73, 186, 198, 251]",0.0,0.0,0.06,0.89,0.01,0.0
1 angry0 box0 disgust0 fear0 happy0 neutral0 sad0 surprise0
2 0.19 [72, 192, 207, 269] 0.0 0.06 0.0 0.58 0.17 0.0
3 0.14 [72, 192, 208, 269] 0.0 0.04 0.0 0.68 0.14 0.0
4 0.2 [71, 191, 208, 271] 0.0 0.02 0.0 0.62 0.15 0.0
5 0.24 [71, 192, 208, 267] 0.0 0.02 0.0 0.59 0.15 0.0
6 0.28 [72, 193, 206, 266] 0.0 0.02 0.0 0.56 0.13 0.0
7 0.33 [69, 189, 209, 265] 0.0 0.04 0.0 0.51 0.11 0.0
8 0.3 [70, 191, 205, 262] 0.0 0.02 0.0 0.56 0.12 0.0
9 0.25 [69, 190, 206, 263] 0.0 0.02 0.0 0.62 0.11 0.0
10 0.22 [69, 190, 206, 262] 0.0 0.02 0.0 0.62 0.14 0.0
11 0.17 [73, 189, 203, 264] 0.0 0.02 0.0 0.67 0.14 0.0
12 0.17 [72, 188, 203, 262] 0.0 0.02 0.0 0.7 0.1 0.0
13 0.16 [72, 187, 202, 263] 0.0 0.02 0.0 0.7 0.11 0.0
14 0.12 [72, 187, 204, 263] 0.0 0.02 0.0 0.78 0.08 0.0
15 0.1 [72, 186, 204, 263] 0.0 0.03 0.0 0.79 0.09 0.0
16 0.15 [70, 185, 206, 264] 0.0 0.03 0.0 0.7 0.12 0.0
17 0.18 [69, 186, 206, 260] 0.0 0.04 0.0 0.64 0.14 0.0
18 0.14 [70, 188, 205, 255] 0.0 0.04 0.0 0.67 0.14 0.0
19 0.17 [71, 189, 204, 254] 0.0 0.03 0.0 0.63 0.17 0.0
20 0.36 [61, 180, 213, 265] 0.0 0.01 0.0 0.54 0.08 0.0
21 0.24 [71, 189, 203, 257] 0.0 0.03 0.0 0.59 0.14 0.0
22 0.15 [68, 188, 204, 258] 0.0 0.03 0.0 0.64 0.17 0.0
23 0.24 [64, 181, 210, 265] 0.0 0.03 0.0 0.49 0.24 0.0
24 0.35 [64, 179, 209, 264] 0.0 0.02 0.0 0.5 0.13 0.0
25 0.44 [69, 182, 205, 255] 0.0 0.02 0.0 0.31 0.23 0.0
26 0.41 [71, 182, 202, 252] 0.0 0.01 0.0 0.27 0.3 0.0
27 0.44 [71, 180, 201, 251] 0.0 0.01 0.0 0.3 0.25 0.0
28 0.41 [65, 177, 207, 258] 0.0 0.02 0.0 0.34 0.23 0.0
29 0.48 [73, 181, 200, 248] 0.0 0.03 0.0 0.2 0.29 0.0
30 0.4 [72, 179, 201, 251] 0.0 0.05 0.0 0.21 0.34 0.0
31 0.39 [61, 172, 208, 257] 0.0 0.03 0.0 0.42 0.15 0.0
32 0.4 [63, 174, 205, 252] 0.0 0.04 0.0 0.37 0.19 0.0
33 0.49 [64, 175, 205, 252] 0.0 0.04 0.0 0.22 0.25 0.0
34 0.42 [65, 174, 204, 251] 0.0 0.03 0.0 0.32 0.21 0.0
35 0.34 [64, 175, 204, 250] 0.0 0.06 0.0 0.33 0.27 0.0
36 0.42 [64, 174, 205, 251] 0.0 0.05 0.0 0.28 0.24 0.0
37 0.45 [64, 174, 205, 251] 0.0 0.05 0.0 0.26 0.23 0.0
38 0.48 [63, 173, 206, 253] 0.0 0.04 0.0 0.27 0.2 0.01
39 0.38 [62, 173, 207, 255] 0.0 0.04 0.0 0.36 0.21 0.01
40 0.35 [62, 173, 207, 257] 0.0 0.03 0.0 0.38 0.22 0.01
41 0.43 [66, 176, 202, 254] 0.0 0.05 0.01 0.26 0.25 0.0
42 0.33 [65, 176, 206, 258] 0.0 0.04 0.0 0.47 0.15 0.0
43 0.17 [65, 175, 207, 260] 0.0 0.05 0.01 0.64 0.14 0.0
44 0.22 [65, 174, 209, 263] 0.0 0.02 0.01 0.6 0.15 0.0
45 0.14 [66, 177, 208, 259] 0.0 0.02 0.07 0.67 0.11 0.0
46 0.07 [67, 173, 210, 262] 0.0 0.01 0.35 0.52 0.05 0.0
47 0.02 [67, 174, 209, 262] 0.0 0.01 0.66 0.29 0.03 0.0
48 0.0 [68, 175, 208, 260] 0.0 0.0 0.89 0.1 0.0 0.0
49 0.0 [72, 176, 202, 256] 0.0 0.0 0.93 0.07 0.0 0.0
50 0.0 [69, 174, 204, 259] 0.0 0.0 0.97 0.03 0.0 0.0
51 0.0 [65, 173, 208, 266] 0.0 0.0 0.99 0.01 0.0 0.0
52 0.0 [61, 172, 213, 269] 0.0 0.0 1.0 0.0 0.0 0.0
53 0.0 [61, 174, 211, 270] 0.0 0.0 1.0 0.0 0.0 0.0
54 0.0 [63, 174, 210, 269] 0.0 0.0 1.0 0.0 0.0 0.0
55 0.0 [63, 174, 210, 270] 0.0 0.0 1.0 0.0 0.0 0.0
56 0.0 [63, 175, 213, 271] 0.0 0.0 1.0 0.0 0.0 0.0
57 0.0 [64, 174, 209, 271] 0.0 0.0 0.99 0.0 0.0 0.0
58 0.0 [64, 173, 210, 271] 0.0 0.0 0.99 0.01 0.0 0.0
59 0.0 [66, 174, 209, 271] 0.0 0.0 0.99 0.01 0.0 0.0
60 0.0 [67, 174, 208, 270] 0.0 0.0 0.99 0.01 0.0 0.0
61 0.0 [66, 173, 211, 271] 0.0 0.0 0.99 0.01 0.0 0.0
62 0.0 [62, 172, 224, 271] 0.0 0.0 1.0 0.0 0.0 0.0
63 0.0 [65, 171, 216, 276] 0.0 0.0 0.99 0.01 0.0 0.0
64 0.0 [62, 172, 224, 270] 0.0 0.0 1.0 0.0 0.0 0.0
65 0.0 [60, 172, 224, 272] 0.0 0.0 0.99 0.01 0.0 0.0
66 0.0 [67, 173, 214, 274] 0.0 0.0 0.99 0.01 0.0 0.0
67 0.0 [60, 172, 223, 272] 0.0 0.0 1.0 0.0 0.0 0.0
68 0.0 [61, 171, 224, 270] 0.0 0.0 1.0 0.0 0.0 0.0
69 0.0 [62, 172, 223, 270] 0.0 0.0 1.0 0.0 0.0 0.0
70 0.0 [66, 170, 213, 275] 0.0 0.0 0.99 0.01 0.0 0.0
71 0.0 [59, 171, 225, 270] 0.0 0.0 1.0 0.0 0.0 0.0
72 0.0 [61, 172, 223, 269] 0.0 0.0 1.0 0.0 0.0 0.0
73 0.0 [60, 172, 223, 270] 0.0 0.0 1.0 0.0 0.0 0.0
74 0.0 [61, 171, 223, 270] 0.0 0.0 1.0 0.0 0.0 0.0
75 0.0 [64, 165, 216, 280] 0.0 0.0 0.99 0.01 0.0 0.0
76 0.0 [60, 170, 222, 271] 0.0 0.0 0.99 0.01 0.0 0.0
77 0.0 [63, 172, 221, 270] 0.0 0.0 1.0 0.0 0.0 0.0
78 0.0 [66, 174, 209, 269] 0.0 0.0 0.98 0.02 0.0 0.0
79 0.0 [70, 178, 211, 264] 0.0 0.0 0.93 0.07 0.0 0.0
80 0.01 [71, 181, 209, 266] 0.0 0.0 0.58 0.4 0.0 0.0
81 0.03 [73, 186, 198, 251] 0.0 0.0 0.06 0.89 0.01 0.0