Update visualization.py
Browse files- visualization.py +1 -1
visualization.py
CHANGED
|
@@ -253,7 +253,7 @@ def create_heatmap(t, mse_embeddings, mse_posture, mse_voice, video_fps, total_f
|
|
| 253 |
ax.set_yticklabels(['Voice', 'Posture', 'Face'], fontsize=7)
|
| 254 |
ax.set_xticks([])
|
| 255 |
|
| 256 |
-
ax.axvline(x=frame_count, color='black', linewidth=
|
| 257 |
|
| 258 |
plt.tight_layout(pad=0.5)
|
| 259 |
|
|
|
|
| 253 |
ax.set_yticklabels(['Voice', 'Posture', 'Face'], fontsize=7)
|
| 254 |
ax.set_xticks([])
|
| 255 |
|
| 256 |
+
ax.axvline(x=frame_count, color='black', linewidth=2)
|
| 257 |
|
| 258 |
plt.tight_layout(pad=0.5)
|
| 259 |
|