Update visualization.py
Browse files- visualization.py +1 -1
visualization.py
CHANGED
|
@@ -236,7 +236,7 @@ def create_heatmap(t, mse_embeddings, mse_posture, mse_voice, video_fps, total_f
|
|
| 236 |
ax.set_yticklabels(['Voice', 'Posture', 'Face'], fontsize=7)
|
| 237 |
ax.set_xticks([])
|
| 238 |
|
| 239 |
-
ax.axvline(x=frame_count, color='black', linewidth=
|
| 240 |
|
| 241 |
plt.tight_layout(pad=0.5)
|
| 242 |
|
|
|
|
| 236 |
ax.set_yticklabels(['Voice', 'Posture', 'Face'], fontsize=7)
|
| 237 |
ax.set_xticks([])
|
| 238 |
|
| 239 |
+
ax.axvline(x=frame_count, color='black', linewidth=2)
|
| 240 |
|
| 241 |
plt.tight_layout(pad=0.5)
|
| 242 |
|