fixed failure case

Author: michalfaber

demo.ipynb

@@ -3,7 +3,9 @@
   {
    "cell_type": "code",
    "execution_count": 1,
-   "metadata": {},
+   "metadata": {
+    "collapsed": false
+   },
    "outputs": [
     {
      "name": "stderr",
@@ -202,7 +204,9 @@
   {
    "cell_type": "code",
    "execution_count": 20,
-   "metadata": {},
+   "metadata": {
+    "collapsed": false
+   },
    "outputs": [
     {
      "data": {
@@ -261,7 +265,9 @@
   {
    "cell_type": "code",
    "execution_count": 22,
-   "metadata": {},
+   "metadata": {
+    "collapsed": false
+   },
    "outputs": [
     {
      "name": "stdout",
@@ -383,7 +389,9 @@
   {
    "cell_type": "code",
    "execution_count": 23,
-   "metadata": {},
+   "metadata": {
+    "collapsed": false
+   },
    "outputs": [
     {
      "data": {
@@ -417,7 +425,9 @@
   {
    "cell_type": "code",
    "execution_count": 25,
-   "metadata": {},
+   "metadata": {
+    "collapsed": false
+   },
    "outputs": [
     {
      "data": {
@@ -536,6 +546,9 @@
     "            for j in range(nB):\n",
     "                vec = np.subtract(candB[j][:2], candA[i][:2])\n",
     "                norm = math.sqrt(vec[0]*vec[0] + vec[1]*vec[1])\n",
+    "                # failure case when 2 body parts overlaps\n",
+    "                if norm == 0:\n",
+    "                    continue\n",
     "                vec = np.divide(vec, norm)\n",
     "                \n",
     "                startend = list(zip(np.linspace(candA[i][0], candB[j][0], num=mid_num), \\\n",
@@ -644,7 +657,9 @@
   {
    "cell_type": "code",
    "execution_count": 31,
-   "metadata": {},
+   "metadata": {
+    "collapsed": false
+   },
    "outputs": [
     {
      "data": {
@@ -688,7 +703,9 @@
   {
    "cell_type": "code",
    "execution_count": 32,
-   "metadata": {},
+   "metadata": {
+    "collapsed": false
+   },
    "outputs": [
     {
      "data": {
@@ -739,7 +756,7 @@
  "metadata": {
   "anaconda-cloud": {},
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "Python [default]",
    "language": "python",
    "name": "python3"
   },
@@ -753,9 +770,9 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.1"
+   "version": "3.5.2"
   }
  },
  "nbformat": 4,
- "nbformat_minor": 1
+ "nbformat_minor": 0
 }

demo_image.py

@@ -106,6 +106,9 @@ def process (input_image, params, model_params):
                 for j in range(nB):
                     vec = np.subtract(candB[j][:2], candA[i][:2])
                     norm = math.sqrt(vec[0] * vec[0] + vec[1] * vec[1])
+                    # failure case when 2 body parts overlaps
+                    if norm == 0:
+                        continue
                     vec = np.divide(vec, norm)
 
                     startend = list(zip(np.linspace(candA[i][0], candB[j][0], num=mid_num), \