Update notebook.

Author: vinnik-dmitry07

debug.ipynb

@@ -7305,27 +7305,30 @@
     "\n",
     "\n",
     "def calc_hand_data(iv):\n",
-    "    origin = to_vec(calculate_origin(iv.hmap[0][..., 9]))\n",
+    "    origin = to_vec(calc_origin(iv.hmap[0][..., 9]))\n",
     "#     origin = to_vec([iv.uv[9][1] / 32, 1 - (iv.uv[9][0]) / 32])\n",
     "    \n",
     "#     UV instead of XY\n",
     "#     joints = list([to_vec([*(uv[::-1]/32), xyz[2]]) for xyz, uv in zip(iv.xyz, iv.uv)])\n",
     "#     joints = list([to_vec([*(uv[::-1]/32), xyz[2]]) for xyz, uv in zip(iv.xyz, xy)])\n",
     "    \n",
     "    joints = list([to_vec(xyz) for xyz in iv.xyz.tolist()])  # tolist, because 0mq can't work with np.float\n",
-    "    \n",
-    "    dists_x = np.array([32 *\n",
-    "                        np.linalg.norm(iv.xyz[i[1:], :2] - iv.xyz[i[:-1], :2]) /\n",
-    "                        np.linalg.norm(iv.uv[i[1:]] - iv.uv[i[:-1]])\n",
-    "                        for i in indexes_x])\n",
     "\n",
-    "    dists_y = np.array([32 * \n",
-    "                        np.linalg.norm(iv.xyz[i[1:], :2] - iv.xyz[i[:-1], :2]) /\n",
-    "                        np.linalg.norm(iv.uv[i[1:]] - iv.uv[i[:-1]])\n",
-    "                        for i in indexes_y])\n",
+    "    def calc_dists(indexes):\n",
+    "        return np.array([\n",
+    "            np.linalg.norm(iv.xyz[i[1:], :2] - iv.xyz[i[:-1], :2]) * 32 /\n",
+    "            np.linalg.norm(iv.uv[i[1:]] - iv.uv[i[:-1]])\n",
+    "            for i in indexes\n",
+    "        ])\n",
+    "\n",
+    "    dists_x = filter_nan_inf(calc_dists(indexes_x))\n",
+    "    dists_y = filter_nan_inf(calc_dists(indexes_y))\n",
+    "\n",
+    "    if dists_x.size == 0 or dists_y.size == 0:\n",
+    "        return None\n",
     "\n",
-    "    dist_x = np.mean(filter_nan_inf(dists_x))\n",
-    "    dist_y = np.mean(filter_nan_inf(dists_y))\n",
+    "    dist_x = np.mean(dists_x)\n",
+    "    dist_y = np.mean(dists_y)\n",
     "    \n",
     "#     palm_size = sum(np.linalg.norm(iv.xyz[0] - iv.xyz[i]) for i in [1, 5, 9, 13, 17])\n",
     "    \n",
@@ -7336,8 +7339,8 @@
     "    return origin, joints, dist_x, dist_y, vert\n",
     "\n",
     "\n",
-    "data_originL, data_jointsL, data_distXL, data_distYL, data_palmSizeL, data_vertL = calc_hand_data(ivl)\n",
-    "data_originR, data_jointsR, data_distXR, data_distYR, data_palmSizeR, data_vertR = calc_hand_data(ivr)\n",
+    "data_originL, data_jointsL, data_distXL, data_distYL, data_vertL = calc_hand_data(ivl)\n",
+    "data_originR, data_jointsR, data_distXR, data_distYR, data_vertR = calc_hand_data(ivr)\n",
     "\n",
     "scaleXL = data_vertL * data_distXL + (1 - data_vertL) * data_distYL\n",
     "scaleYL = data_vertL * data_distYL + (1 - data_vertL) * data_distXL\n",