[Fix] Fix issue kennymckormick#153 (bug of heatmap visualization note…

…book) (kennymckormick#154)

demo/visualize_heatmap_volume.ipynb

@@ -11,6 +11,7 @@
     "import cv2\n",
     "import os.path as osp\n",
     "import decord\n",
+    "import copy as cp\n",
     "import numpy as np\n",
     "import matplotlib.pyplot as plt\n",
     "import urllib\n",
@@ -211,7 +212,7 @@
    "outputs": [],
    "source": [
     "# Visualize Skeleton\n",
-    "vis_frames = vis_skeleton(vid_path, anno, gym_categories[anno['label']])\n",
+    "vis_frames = vis_skeleton(vid_path, cp.deepcopy(anno), gym_categories[anno['label']])\n",
     "vid = mpy.ImageSequenceClip(vis_frames, fps=24)\n",
     "vid.ipython_display()"
    ]
@@ -223,7 +224,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "keypoint_heatmap = get_pseudo_heatmap(anno)\n",
+    "keypoint_heatmap = get_pseudo_heatmap(cp.deepcopy(anno))\n",
     "keypoint_mapvis = vis_heatmaps(keypoint_heatmap)\n",
     "keypoint_mapvis = [add_label(f, gym_categories[anno['label']]) for f in keypoint_mapvis]\n",
     "vid = mpy.ImageSequenceClip(keypoint_mapvis, fps=24)\n",
@@ -237,7 +238,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "limb_heatmap = get_pseudo_heatmap(anno, 'limb')\n",
+    "limb_heatmap = get_pseudo_heatmap(cp.deepcopy(anno), 'limb')\n",
     "limb_mapvis = vis_heatmaps(limb_heatmap)\n",
     "limb_mapvis = [add_label(f, gym_categories[anno['label']]) for f in limb_mapvis]\n",
     "vid = mpy.ImageSequenceClip(limb_mapvis, fps=24)\n",
@@ -310,7 +311,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "vis_frames = vis_skeleton(vid_path, anno, ntu_categories[anno['label']])\n",
+    "vis_frames = vis_skeleton(vid_path, cp.deepcopy(anno), ntu_categories[anno['label']])\n",
     "vid = mpy.ImageSequenceClip(vis_frames, fps=24)\n",
     "vid.ipython_display()"
    ]
@@ -322,9 +323,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "keypoint_heatmap = get_pseudo_heatmap(anno)\n",
+    "keypoint_heatmap = get_pseudo_heatmap(cp.deepcopy(anno))\n",
     "keypoint_mapvis = vis_heatmaps(keypoint_heatmap)\n",
-    "keypoint_mapvis = [add_label(f, gym_categories[anno['label']]) for f in keypoint_mapvis]\n",
+    "keypoint_mapvis = [add_label(f, ntu_categories[anno['label']]) for f in keypoint_mapvis]\n",
     "vid = mpy.ImageSequenceClip(keypoint_mapvis, fps=24)\n",
     "vid.ipython_display()"
    ]
@@ -336,9 +337,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "limb_heatmap = get_pseudo_heatmap(anno, 'limb')\n",
+    "limb_heatmap = get_pseudo_heatmap(cp.deepcopy(anno), 'limb')\n",
     "limb_mapvis = vis_heatmaps(limb_heatmap)\n",
-    "limb_mapvis = [add_label(f, gym_categories[anno['label']]) for f in limb_mapvis]\n",
+    "limb_mapvis = [add_label(f, ntu_categories[anno['label']]) for f in limb_mapvis]\n",
     "vid = mpy.ImageSequenceClip(limb_mapvis, fps=24)\n",
     "vid.ipython_display()"
    ]