obj_det.ipynb

{
 "metadata": {
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.10"
  },
  "orig_nbformat": 2,
  "kernelspec": {
   "name": "python3710jvsc74a57bd08158935654b77d7cd29582f0e5284d9414d7156c4d4d118169d2fe9f6b52ce92",
   "display_name": "Python 3.7.10 64-bit ('objdet': conda)"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2,
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "WARNING:tensorflow:No training configuration found in the save file, so the model was *not* compiled. Compile it manually.\n",
      "Processing Frame :  1\n",
      "Processing Frame :  2\n",
      "Processing Frame :  3\n",
      "Processing Frame :  4\n",
      "Processing Frame :  5\n",
      "Processing Frame :  6\n",
      "Processing Frame :  7\n",
      "Processing Frame :  8\n",
      "Processing Frame :  9\n",
      "Processing Frame :  10\n",
      "Processing Frame :  11\n",
      "Processing Frame :  12\n",
      "Processing Frame :  13\n",
      "Processing Frame :  14\n",
      "Processing Frame :  15\n",
      "Processing Frame :  16\n",
      "Processing Frame :  17\n",
      "Processing Frame :  18\n",
      "Processing Frame :  19\n",
      "Processing Frame :  20\n",
      "Processing Frame :  21\n",
      "Processing Frame :  22\n",
      "Processing Frame :  23\n",
      "Processing Frame :  24\n",
      "Processing Frame :  25\n",
      "Processing Frame :  26\n",
      "Processing Frame :  27\n",
      "Processing Frame :  28\n",
      "Processing Frame :  29\n",
      "Processing Frame :  30\n",
      "Processing Frame :  31\n",
      "Processing Frame :  32\n",
      "Processing Frame :  33\n",
      "Processing Frame :  34\n",
      "Processing Frame :  35\n",
      "Processing Frame :  36\n",
      "Processing Frame :  37\n",
      "Processing Frame :  38\n",
      "Processing Frame :  39\n",
      "Processing Frame :  40\n",
      "Processing Frame :  41\n",
      "Processing Frame :  42\n",
      "Processing Frame :  43\n",
      "Processing Frame :  44\n",
      "Processing Frame :  45\n",
      "Processing Frame :  46\n",
      "Processing Frame :  47\n",
      "Processing Frame :  48\n",
      "Processing Frame :  49\n",
      "Processing Frame :  50\n",
      "Processing Frame :  51\n",
      "Processing Frame :  52\n",
      "Processing Frame :  53\n",
      "Processing Frame :  54\n",
      "Processing Frame :  55\n",
      "Processing Frame :  56\n",
      "Processing Frame :  57\n",
      "Processing Frame :  58\n",
      "Processing Frame :  59\n",
      "Processing Frame :  60\n",
      "Processing Frame :  61\n",
      "Processing Frame :  62\n",
      "Processing Frame :  63\n",
      "Processing Frame :  64\n",
      "Processing Frame :  65\n",
      "Processing Frame :  66\n",
      "Processing Frame :  67\n",
      "Processing Frame :  68\n",
      "Processing Frame :  69\n",
      "Processing Frame :  70\n",
      "Processing Frame :  71\n",
      "Processing Frame :  72\n",
      "Processing Frame :  73\n",
      "Processing Frame :  74\n",
      "Processing Frame :  75\n",
      "Processing Frame :  76\n",
      "Processing Frame :  77\n",
      "Processing Frame :  78\n",
      "Processing Frame :  79\n",
      "Processing Frame :  80\n",
      "Processing Frame :  81\n",
      "Processing Frame :  82\n",
      "Processing Frame :  83\n",
      "Processing Frame :  84\n",
      "Processing Frame :  85\n",
      "Processing Frame :  86\n",
      "Processing Frame :  87\n",
      "Processing Frame :  88\n",
      "Processing Frame :  89\n",
      "Processing Frame :  90\n",
      "Processing Frame :  91\n",
      "Processing Frame :  92\n",
      "Processing Frame :  93\n",
      "Processing Frame :  94\n",
      "Processing Frame :  95\n",
      "Processing Frame :  96\n",
      "Processing Frame :  97\n",
      "Processing Frame :  98\n",
      "Processing Frame :  99\n",
      "Processing Frame :  100\n",
      "Processing Frame :  101\n",
      "Processing Frame :  102\n",
      "Processing Frame :  103\n",
      "Processing Frame :  104\n",
      "Processing Frame :  105\n",
      "Processing Frame :  106\n",
      "Processing Frame :  107\n",
      "Processing Frame :  108\n",
      "Processing Frame :  109\n",
      "Processing Frame :  110\n",
      "Processing Frame :  111\n",
      "Processing Frame :  112\n",
      "Processing Frame :  113\n",
      "Processing Frame :  114\n",
      "Processing Frame :  115\n",
      "Processing Frame :  116\n",
      "Processing Frame :  117\n",
      "Processing Frame :  118\n",
      "Processing Frame :  119\n",
      "Processing Frame :  120\n",
      "Processing Frame :  121\n",
      "Processing Frame :  122\n",
      "Processing Frame :  123\n",
      "Processing Frame :  124\n",
      "Processing Frame :  125\n",
      "Processing Frame :  126\n",
      "Processing Frame :  127\n",
      "Processing Frame :  128\n",
      "Processing Frame :  129\n",
      "Processing Frame :  130\n",
      "Processing Frame :  131\n",
      "Processing Frame :  132\n",
      "Processing Frame :  133\n",
      "Processing Frame :  134\n",
      "Processing Frame :  135\n",
      "Processing Frame :  136\n",
      "Processing Frame :  137\n",
      "Processing Frame :  138\n",
      "Processing Frame :  139\n",
      "Processing Frame :  140\n",
      "Processing Frame :  141\n",
      "Processing Frame :  142\n",
      "Processing Frame :  143\n",
      "Processing Frame :  144\n",
      "Processing Frame :  145\n",
      "Processing Frame :  146\n",
      "Processing Frame :  147\n",
      "Processing Frame :  148\n",
      "Processing Frame :  149\n",
      "Processing Frame :  150\n",
      "Processing Frame :  151\n",
      "Processing Frame :  152\n",
      "Processing Frame :  153\n",
      "Processing Frame :  154\n",
      "Processing Frame :  155\n",
      "Processing Frame :  156\n",
      "Processing Frame :  157\n",
      "Processing Frame :  158\n",
      "Processing Frame :  159\n",
      "Processing Frame :  160\n",
      "Processing Frame :  161\n",
      "Processing Frame :  162\n",
      "Processing Frame :  163\n",
      "Processing Frame :  164\n",
      "Processing Frame :  165\n",
      "Processing Frame :  166\n",
      "Processing Frame :  167\n",
      "Processing Frame :  168\n",
      "Processing Frame :  169\n",
      "Processing Frame :  170\n",
      "Processing Frame :  171\n",
      "Processing Frame :  172\n",
      "Processing Frame :  173\n",
      "Processing Frame :  174\n",
      "Processing Frame :  175\n",
      "Processing Frame :  176\n",
      "Processing Frame :  177\n",
      "Processing Frame :  178\n",
      "Processing Frame :  179\n",
      "Processing Frame :  180\n",
      "Processing Frame :  181\n",
      "Processing Frame :  182\n",
      "Processing Frame :  183\n",
      "Processing Frame :  184\n",
      "Processing Frame :  185\n",
      "Processing Frame :  186\n",
      "Processing Frame :  187\n",
      "Processing Frame :  188\n",
      "Processing Frame :  189\n",
      "Processing Frame :  190\n",
      "Processing Frame :  191\n",
      "Processing Frame :  192\n",
      "Processing Frame :  193\n",
      "Processing Frame :  194\n",
      "Processing Frame :  195\n",
      "Processing Frame :  196\n",
      "Processing Frame :  197\n",
      "Processing Frame :  198\n",
      "Processing Frame :  199\n",
      "Processing Frame :  200\n",
      "Processing Frame :  201\n",
      "Processing Frame :  202\n",
      "Processing Frame :  203\n",
      "Processing Frame :  204\n",
      "Processing Frame :  205\n",
      "Processing Frame :  206\n",
      "Processing Frame :  207\n",
      "Processing Frame :  208\n",
      "Processing Frame :  209\n",
      "Processing Frame :  210\n",
      "Processing Frame :  211\n",
      "Processing Frame :  212\n",
      "Processing Frame :  213\n",
      "Processing Frame :  214\n",
      "Processing Frame :  215\n",
      "Processing Frame :  216\n",
      "Processing Frame :  217\n",
      "Processing Frame :  218\n",
      "Processing Frame :  219\n",
      "Processing Frame :  220\n",
      "Processing Frame :  221\n",
      "Processing Frame :  222\n",
      "Processing Frame :  223\n",
      "Processing Frame :  224\n",
      "Processing Frame :  225\n",
      "Processing Frame :  226\n",
      "Processing Frame :  227\n",
      "Processing Frame :  228\n",
      "Processing Frame :  229\n",
      "Processing Frame :  230\n",
      "Processing Frame :  231\n",
      "Processing Frame :  232\n",
      "Processing Frame :  233\n",
      "Processing Frame :  234\n",
      "Processing Frame :  235\n",
      "Processing Frame :  236\n",
      "Processing Frame :  237\n",
      "Processing Frame :  238\n",
      "Processing Frame :  239\n",
      "Processing Frame :  240\n",
      "Processing Frame :  241\n",
      "Processing Frame :  242\n",
      "Processing Frame :  243\n",
      "Processing Frame :  244\n",
      "Processing Frame :  245\n",
      "Processing Frame :  246\n",
      "Processing Frame :  247\n",
      "Processing Frame :  248\n",
      "Processing Frame :  249\n",
      "Processing Frame :  250\n",
      "Processing Frame :  251\n",
      "Processing Frame :  252\n",
      "Processing Frame :  253\n",
      "Processing Frame :  254\n",
      "Processing Frame :  255\n",
      "Processing Frame :  256\n",
      "Processing Frame :  257\n",
      "Processing Frame :  258\n",
      "Processing Frame :  259\n",
      "Processing Frame :  260\n",
      "Processing Frame :  261\n",
      "Processing Frame :  262\n",
      "Processing Frame :  263\n",
      "Processing Frame :  264\n",
      "Processing Frame :  265\n",
      "Processing Frame :  266\n",
      "Processing Frame :  267\n",
      "Processing Frame :  268\n",
      "Processing Frame :  269\n",
      "Processing Frame :  270\n",
      "Processing Frame :  271\n",
      "Processing Frame :  272\n",
      "Processing Frame :  273\n",
      "Processing Frame :  274\n",
      "Processing Frame :  275\n",
      "Processing Frame :  276\n",
      "Processing Frame :  277\n",
      "Processing Frame :  278\n",
      "Processing Frame :  279\n",
      "Processing Frame :  280\n",
      "Processing Frame :  281\n",
      "Processing Frame :  282\n",
      "Processing Frame :  283\n",
      "Processing Frame :  284\n",
      "Processing Frame :  285\n",
      "Processing Frame :  286\n",
      "Processing Frame :  287\n",
      "Processing Frame :  288\n",
      "Processing Frame :  289\n",
      "Processing Frame :  290\n",
      "Processing Frame :  291\n",
      "Processing Frame :  292\n",
      "Processing Frame :  293\n",
      "Processing Frame :  294\n",
      "Processing Frame :  295\n",
      "Processing Frame :  296\n",
      "Processing Frame :  297\n",
      "Processing Frame :  298\n",
      "Processing Frame :  299\n",
      "Processing Frame :  300\n",
      "Processing Frame :  301\n",
      "Processing Frame :  302\n",
      "Processing Frame :  303\n",
      "Processing Frame :  304\n",
      "Processing Frame :  305\n",
      "Processing Frame :  306\n",
      "Processing Frame :  307\n",
      "Processing Frame :  308\n",
      "Processing Frame :  309\n",
      "Processing Frame :  310\n",
      "Processing Frame :  311\n",
      "Processing Frame :  312\n",
      "Processing Frame :  313\n",
      "Processing Frame :  314\n",
      "Processing Frame :  315\n",
      "Processing Frame :  316\n",
      "Processing Frame :  317\n",
      "Processing Frame :  318\n",
      "Processing Frame :  319\n",
      "Processing Frame :  320\n",
      "Processing Frame :  321\n",
      "Processing Frame :  322\n",
      "Processing Frame :  323\n",
      "Processing Frame :  324\n",
      "Processing Frame :  325\n",
      "Processing Frame :  326\n",
      "Processing Frame :  327\n",
      "Processing Frame :  328\n",
      "Processing Frame :  329\n",
      "Processing Frame :  330\n",
      "Processing Frame :  331\n",
      "Processing Frame :  332\n",
      "Processing Frame :  333\n",
      "Processing Frame :  334\n",
      "Processing Frame :  335\n",
      "Processing Frame :  336\n",
      "Processing Frame :  337\n",
      "Processing Frame :  338\n",
      "Processing Frame :  339\n",
      "Processing Frame :  340\n",
      "Processing Frame :  341\n",
      "Processing Frame :  342\n",
      "Processing Frame :  343\n",
      "Processing Frame :  344\n",
      "Processing Frame :  345\n",
      "Processing Frame :  346\n",
      "Processing Frame :  347\n",
      "Processing Frame :  348\n",
      "Processing Frame :  349\n",
      "Processing Frame :  350\n",
      "Processing Frame :  351\n",
      "Processing Frame :  352\n",
      "Processing Frame :  353\n",
      "Processing Frame :  354\n",
      "Processing Frame :  355\n",
      "Processing Frame :  356\n",
      "Processing Frame :  357\n",
      "Processing Frame :  358\n",
      "Processing Frame :  359\n",
      "Processing Frame :  360\n",
      "Processing Frame :  361\n",
      "Processing Frame :  362\n",
      "Processing Frame :  363\n",
      "Processing Frame :  364\n",
      "Processing Frame :  365\n",
      "Processing Frame :  366\n",
      "Processing Frame :  367\n",
      "Processing Frame :  368\n",
      "Processing Frame :  369\n",
      "Processing Frame :  370\n",
      "Processing Frame :  371\n",
      "Processing Frame :  372\n",
      "Processing Frame :  373\n",
      "Processing Frame :  374\n",
      "Processing Frame :  375\n",
      "Processing Frame :  376\n",
      "Processing Frame :  377\n",
      "Processing Frame :  378\n",
      "Processing Frame :  379\n",
      "Processing Frame :  380\n",
      "Processing Frame :  381\n",
      "Processing Frame :  382\n",
      "Processing Frame :  383\n",
      "Processing Frame :  384\n",
      "Processing Frame :  385\n",
      "Processing Frame :  386\n",
      "Processing Frame :  387\n",
      "Processing Frame :  388\n",
      "Processing Frame :  389\n",
      "Processing Frame :  390\n",
      "Processing Frame :  391\n",
      "Processing Frame :  392\n",
      "Processing Frame :  393\n",
      "Processing Frame :  394\n",
      "Processing Frame :  395\n",
      "Processing Frame :  396\n",
      "Processing Frame :  397\n",
      "Processing Frame :  398\n",
      "Processing Frame :  399\n",
      "Processing Frame :  400\n",
      "Processing Frame :  401\n",
      "Processing Frame :  402\n",
      "Processing Frame :  403\n",
      "Processing Frame :  404\n",
      "Processing Frame :  405\n",
      "Processing Frame :  406\n",
      "Processing Frame :  407\n",
      "Processing Frame :  408\n",
      "Processing Frame :  409\n",
      "Processing Frame :  410\n",
      "Processing Frame :  411\n",
      "Processing Frame :  412\n",
      "Processing Frame :  413\n",
      "Processing Frame :  414\n",
      "Processing Frame :  415\n",
      "Processing Frame :  416\n",
      "Processing Frame :  417\n",
      "Processing Frame :  418\n",
      "Processing Frame :  419\n",
      "Processing Frame :  420\n",
      "Processing Frame :  421\n",
      "Processing Frame :  422\n",
      "Processing Frame :  423\n",
      "Processing Frame :  424\n",
      "Processing Frame :  425\n",
      "Processing Frame :  426\n",
      "Processing Frame :  427\n",
      "Processing Frame :  428\n",
      "Processing Frame :  429\n",
      "Processing Frame :  430\n",
      "Processing Frame :  431\n",
      "Processing Frame :  432\n",
      "Processing Frame :  433\n",
      "Processing Frame :  434\n",
      "Processing Frame :  435\n",
      "Processing Frame :  436\n",
      "Processing Frame :  437\n",
      "Processing Frame :  438\n",
      "Processing Frame :  439\n",
      "Processing Frame :  440\n",
      "Processing Frame :  441\n",
      "Processing Frame :  442\n"
     ]
    }
   ],
   "source": [
    "from imageai.Detection import VideoObjectDetection\n",
    "import os\n",
    "\n",
    "execution_path = os.getcwd()\n",
    "\n",
    "detector = VideoObjectDetection()\n",
    "detector.setModelTypeAsRetinaNet()\n",
    "detector.setModelPath( os.path.join(execution_path , \"checkpoints/resnet50_coco_best_v2.1.0.h5\"))\n",
    "detector.loadModel()\n",
    "custom_objects = detector.CustomObjects(sports_ball=True,baseball_bat=True)\n",
    "\n",
    "video_path = detector.detectObjectsFromVideo(input_file_path=os.path.join(execution_path, \"../data/video.mp4\"),\n",
    "                                output_file_path=os.path.join(execution_path, \"traffic_detected\")\n",
    "                                , frames_per_second=20, log_progress=True)\n",
    "print(video_path)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "tags": [
     "outputPrepend"
    ]
   },
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      " objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1476\n",
      "FOR FRAME  1476\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1477\n",
      "FOR FRAME  1477\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1478\n",
      "FOR FRAME  1478\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1479\n",
      "FOR FRAME  1479\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1480\n",
      "FOR FRAME  1480\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1481\n",
      "FOR FRAME  1481\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1482\n",
      "FOR FRAME  1482\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1483\n",
      "FOR FRAME  1483\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1484\n",
      "FOR FRAME  1484\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1485\n",
      "FOR FRAME  1485\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1486\n",
      "FOR FRAME  1486\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1487\n",
      "FOR FRAME  1487\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1488\n",
      "FOR FRAME  1488\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1489\n",
      "FOR FRAME  1489\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1490\n",
      "FOR FRAME  1490\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1491\n",
      "FOR FRAME  1491\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1492\n",
      "FOR FRAME  1492\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1493\n",
      "FOR FRAME  1493\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1494\n",
      "FOR FRAME  1494\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1495\n",
      "FOR FRAME  1495\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1496\n",
      "FOR FRAME  1496\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1497\n",
      "FOR FRAME  1497\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1498\n",
      "FOR FRAME  1498\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1499\n",
      "FOR FRAME  1499\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1500\n",
      "FOR FRAME  1500\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1501\n",
      "FOR FRAME  1501\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1502\n",
      "FOR FRAME  1502\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1503\n",
      "FOR FRAME  1503\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1504\n",
      "FOR FRAME  1504\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1505\n",
      "FOR FRAME  1505\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1506\n",
      "FOR FRAME  1506\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1507\n",
      "FOR FRAME  1507\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1508\n",
      "FOR FRAME  1508\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1509\n",
      "FOR FRAME  1509\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1510\n",
      "FOR FRAME  1510\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1511\n",
      "FOR FRAME  1511\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1512\n",
      "FOR FRAME  1512\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1513\n",
      "FOR FRAME  1513\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1514\n",
      "FOR FRAME  1514\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1515\n",
      "FOR FRAME  1515\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1516\n",
      "FOR FRAME  1516\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1517\n",
      "FOR FRAME  1517\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1518\n",
      "FOR FRAME  1518\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1519\n",
      "FOR FRAME  1519\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1520\n",
      "FOR FRAME  1520\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1521\n",
      "FOR FRAME  1521\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1522\n",
      "FOR FRAME  1522\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1523\n",
      "FOR FRAME  1523\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1524\n",
      "FOR FRAME  1524\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1525\n",
      "FOR FRAME  1525\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1526\n",
      "FOR FRAME  1526\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1527\n",
      "FOR FRAME  1527\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1528\n",
      "FOR FRAME  1528\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1529\n",
      "FOR FRAME  1529\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1530\n",
      "FOR FRAME  1530\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1531\n",
      "FOR FRAME  1531\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1532\n",
      "FOR FRAME  1532\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1533\n",
      "FOR FRAME  1533\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1534\n",
      "FOR FRAME  1534\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1535\n",
      "FOR FRAME  1535\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1536\n",
      "FOR FRAME  1536\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1537\n",
      "FOR FRAME  1537\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1538\n",
      "FOR FRAME  1538\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1539\n",
      "FOR FRAME  1539\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1540\n",
      "FOR FRAME  1540\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1541\n",
      "FOR FRAME  1541\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1542\n",
      "FOR FRAME  1542\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1543\n",
      "FOR FRAME  1543\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1544\n",
      "FOR FRAME  1544\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1545\n",
      "FOR FRAME  1545\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1546\n",
      "FOR FRAME  1546\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1547\n",
      "FOR FRAME  1547\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1548\n",
      "FOR FRAME  1548\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1549\n",
      "FOR FRAME  1549\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1550\n",
      "FOR FRAME  1550\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1551\n",
      "FOR FRAME  1551\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1552\n",
      "FOR FRAME  1552\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1553\n",
      "FOR FRAME  1553\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1554\n",
      "FOR FRAME  1554\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1555\n",
      "FOR FRAME  1555\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1556\n",
      "FOR FRAME  1556\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1557\n",
      "FOR FRAME  1557\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1558\n",
      "FOR FRAME  1558\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1559\n",
      "FOR FRAME  1559\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1560\n",
      "FOR FRAME  1560\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1561\n",
      "FOR FRAME  1561\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1562\n",
      "FOR FRAME  1562\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1563\n",
      "FOR FRAME  1563\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1564\n",
      "FOR FRAME  1564\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1565\n",
      "FOR FRAME  1565\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1566\n",
      "FOR FRAME  1566\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1567\n",
      "FOR FRAME  1567\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1568\n",
      "FOR FRAME  1568\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1569\n",
      "FOR FRAME  1569\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1570\n",
      "FOR FRAME  1570\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1571\n",
      "FOR FRAME  1571\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1572\n",
      "FOR FRAME  1572\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1573\n",
      "FOR FRAME  1573\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1574\n",
      "FOR FRAME  1574\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1575\n",
      "FOR FRAME  1575\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1576\n",
      "FOR FRAME  1576\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1577\n",
      "FOR FRAME  1577\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1578\n",
      "FOR FRAME  1578\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1579\n",
      "FOR FRAME  1579\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1580\n",
      "FOR FRAME  1580\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1581\n",
      "FOR FRAME  1581\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1582\n",
      "FOR FRAME  1582\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1583\n",
      "FOR FRAME  1583\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1584\n",
      "FOR FRAME  1584\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1585\n",
      "FOR FRAME  1585\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1586\n",
      "FOR FRAME  1586\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1587\n",
      "FOR FRAME  1587\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1588\n",
      "FOR FRAME  1588\n",
      "Output for each object :  [{'name': 'sports ball', 'percentage_probability': 55.61046004295349, 'box_points': [675, 18, 702, 49]}]\n",
      "Output count for unique objects :  {'sports ball': 1}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1589\n",
      "FOR FRAME  1589\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1590\n",
      "FOR FRAME  1590\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1591\n",
      "FOR FRAME  1591\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1592\n",
      "FOR FRAME  1592\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1593\n",
      "FOR FRAME  1593\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1594\n",
      "FOR FRAME  1594\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1595\n",
      "FOR FRAME  1595\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1596\n",
      "FOR FRAME  1596\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1597\n",
      "FOR FRAME  1597\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1598\n",
      "FOR FRAME  1598\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1599\n",
      "FOR FRAME  1599\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1600\n",
      "FOR FRAME  1600\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1601\n",
      "FOR FRAME  1601\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1602\n",
      "FOR FRAME  1602\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1603\n",
      "FOR FRAME  1603\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1604\n",
      "FOR FRAME  1604\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1605\n",
      "FOR FRAME  1605\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1606\n",
      "FOR FRAME  1606\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1607\n",
      "FOR FRAME  1607\n",
      "Output for each object :  []\n",
      "Output count for unique objects :  {}\n",
      "------------END OF A FRAME --------------\n",
      "Processing Frame :  1608\n"
     ]
    }
   ],
   "source": [
    "from imageai.Detection import VideoObjectDetection\n",
    "import os\n",
    "\n",
    "boxes = []\n",
    "def forFrame(frame_number, output_array, output_count):\n",
    "    boxes.append(output_array)\n",
    "    print(\"FOR FRAME \" , frame_number)\n",
    "    print(\"Output for each object : \", output_array)\n",
    "    print(\"Output count for unique objects : \", output_count)\n",
    "    print(\"------------END OF A FRAME --------------\")\n",
    "\n",
    "execution_path = os.getcwd()\n",
    "\n",
    "detector = VideoObjectDetection()\n",
    "detector.setModelTypeAsRetinaNet()\n",
    "detector.setModelPath( os.path.join(execution_path , \"checkpoints/resnet50_coco_best_v2.1.0.h5\"))\n",
    "detector.loadModel()\n",
    "\n",
    "custom_objects = detector.CustomObjects(sports_ball=True,baseball_bat=True)\n",
    "\n",
    "video_path = detector.detectCustomObjectsFromVideo(\n",
    "                custom_objects=custom_objects,\n",
    "                input_file_path=os.path.join(execution_path, \"data/red_cricket.avi\"),\n",
    "                output_file_path=os.path.join(execution_path, \"results/bat_ball_det\"),\n",
    "                frames_per_second=2, log_progress=True,\n",
    "                per_frame_function = forFrame\n",
    "                )\n",
    "print(video_path)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 37,
   "metadata": {},
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "[129, [{'name': 'baseball bat', 'percentage_probability': 51.58252716064453, 'box_points': [995, 338, 1204, 533]}]]\n[506, 1964, 2035, 3455, 4256, 5980]\n"
     ]
    }
   ],
   "source": [
    "import json \n",
    "with open('results/detfps60fd1normal.json', 'r') as f:\n",
    "    content = json.load(f)\n",
    "print(content[0])\n",
    "\n",
    "th = 50\n",
    "new_out = []\n",
    "bat =[]\n",
    "cor_bat =[]\n",
    "ball =[]\n",
    "cor_ball =[]\n",
    "lst_name =''\n",
    "pair = False\n",
    "for i in content:\n",
    "    if len(set([j['name'] for j in i[1]]))>1:\n",
    "        if  len(new_out)>0:\n",
    "            if i[0]-new_out[-1]> th:\n",
    "                # if pair!=True:\n",
    "                    # del new_out[-1]\n",
    "                new_out.append(i[0])\n",
    "                name =''\n",
    "                pair =True\n",
    "                bat.append(i[0])\n",
    "                cor_bat.append(i[1][0]['box_points'])\n",
    "                ball.append(i[0])\n",
    "                cor_ball.append(i[1][0]['box_points'])\n",
    "        else:\n",
    "            new_out.append(i[0])\n",
    "    elif i[1][0]['name']=='baseball bat':\n",
    "        bat.append(i[0])\n",
    "        cor_bat.append(i[1][0]['box_points'])\n",
    "    else:\n",
    "        ball.append(i[0])\n",
    "        cor_ball.append(i[1][0]['box_points'])\n",
    "        # ball.append(i[0])\n",
    "\n",
    "        # else:\n",
    "        #     new_out.append(i[0])\n",
    "        #     name =i[1][0]['name']\n",
    "        #     pair =False\n",
    "    \n",
    "            # elif name!=i[1][0]['name']:\n",
    "            #     pair= True\n",
    "        # else:\n",
    "        #     new_out.append(i[0])\n",
    "        #     pair =False\n",
    "        #     lst_name = i[1][0]['name']\n",
    "       \n",
    "    # if len(new_out)>0:\n",
    "    #     if i[0]-new_out[-1]> th:\n",
    "    #         if pair!=True:\n",
    "    #             del new_out[-1]\n",
    "    #         new_out.append(i[0])\n",
    "    #         pair=False\n",
    "    #         name = i[1][0]['name']\n",
    "    #     elif name!=i[1][0]['name']:\n",
    "    #         pair= True\n",
    "    # else:\n",
    "    #     new_out.append(i[0])\n",
    "    #     name =i[1][0]['name']\n",
    "    #     pair =False\n",
    "        \n",
    "            \n",
    "    \n",
    "\n",
    "    \n",
    "\n",
    "print(new_out)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 38,
   "metadata": {},
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "ball\n [496, 497, 498, 499, 500, 501, 502, 503, 504, 505, 508, 513, 514, 515, 516, 517, 518, 519, 523, 524, 529, 530, 533, 621, 622, 631, 632, 1065, 1962, 1963, 1964, 1968, 1969, 1971, 1992, 1993, 1994, 1997, 1998, 1999, 2000, 2004, 2007, 2014, 2015, 2016, 2017, 2018, 2019, 2020, 2021, 2022, 2023, 2026, 2027, 2028, 2030, 2031, 2032, 2033, 2034, 2035, 2036, 2638, 2639, 2640, 2641, 2642, 2643, 2644, 2645, 2646, 2647, 2648, 2649, 2650, 2651, 2652, 2653, 2654, 2655, 2656, 2657, 2658, 2659, 2660, 2661, 2662, 2663, 2664, 2665, 2666, 2667, 2668, 2669, 2670, 2671, 2672, 2673, 2674, 2675, 2676, 2677, 2678, 2679, 2680, 2681, 2682, 2711, 3118, 3119, 3127, 3128, 3129, 3130, 3131, 3132, 3133, 3134, 3135, 3136, 3137, 3138, 3139, 3140, 3141, 3142, 3143, 3144, 3145, 3146, 3147, 3148, 3149, 3150, 3151, 3152, 3153, 3164, 3165, 3167, 3399, 3400, 3455, 4143, 4190, 4191, 4192, 4193, 4232, 4235, 4237, 4241, 4242, 4243, 4244, 4245, 4252, 4256, 5845, 5846, 5847, 5848, 5849, 5850, 5851, 5852, 5853, 5854, 5855, 5857, 5858, 5859, 5860, 5864, 5865, 5940, 5941, 5950, 5979, 5980, 7355, 7356, 7357, 7358, 7359, 7360, 7361, 7362, 7363, 7364, 7365, 7366, 7367, 7368, 7369, 7370, 7371, 7372, 7373, 7374, 7375, 7376, 7377, 7378, 7379, 7380, 7381, 7382, 7384, 7385, 7386, 7387, 7388, 7389, 7390, 7391, 7392, 7393, 7394, 7395, 7396, 7397, 7398, 7400, 7401]\nbat\n [129, 217, 219, 222, 223, 240, 241, 243, 249, 258, 528, 560, 561, 562, 563, 564, 565, 566, 567, 568, 569, 570, 571, 725, 726, 727, 728, 729, 730, 731, 736, 746, 884, 885, 1222, 1223, 1224, 1225, 1226, 1227, 1228, 1229, 1237, 1238, 1239, 1242, 1243, 1244, 1245, 1246, 1305, 1306, 1309, 1310, 1311, 1313, 1314, 1334, 1335, 1338, 1339, 1341, 1342, 1343, 1344, 1345, 1765, 1766, 1770, 1784, 1785, 1786, 1787, 1788, 1789, 1793, 1794, 1797, 1805, 1806, 1807, 1808, 1809, 1810, 1811, 1812, 1813, 1814, 1815, 1816, 1964, 1981, 1982, 1983, 1984, 1985, 1986, 2035, 2043, 2044, 2045, 2046, 2047, 2048, 2049, 2051, 2052, 2053, 2054, 2055, 2057, 2058, 2059, 2061, 2065, 2066, 2067, 2068, 2069, 2070, 2071, 2072, 2073, 2074, 2075, 2076, 2078, 2079, 2081, 2082, 2083, 2084, 2085, 2086, 2087, 2088, 2089, 2090, 2091, 2092, 2093, 2094, 2095, 2096, 2097, 2098, 2099, 2100, 2102, 2103, 2104, 2105, 2106, 2107, 2108, 2109, 2110, 2111, 2112, 2113, 2114, 2115, 2116, 2117, 2118, 2119, 2120, 2121, 2122, 2123, 2124, 2125, 2126, 2127, 2404, 2406, 2409, 2424, 2516, 2517, 2526, 2527, 2528, 2576, 2764, 2765, 2766, 2767, 2768, 2769, 2770, 2771, 2772, 2775, 2892, 2893, 2895, 2896, 2897, 2898, 2899, 2900, 2901, 3419, 3438, 3439, 3444, 3445, 3449, 3455, 3458, 3488, 3539, 3542, 3543, 3545, 3546, 3547, 3551, 3555, 3556, 3562, 3583, 3806, 3944, 3949, 3950, 3964, 3965, 3968, 3970, 3971, 3972, 3973, 3983, 3984, 3985, 4183, 4197, 4200, 4201, 4202, 4203, 4204, 4205, 4206, 4207, 4208, 4209, 4210, 4211, 4212, 4213, 4214, 4215, 4216, 4217, 4218, 4219, 4220, 4221, 4222, 4223, 4255, 4256, 4257, 4261, 4262, 4264, 4265, 4266, 4267, 4268, 4273, 4274, 4275, 4276, 4277, 4278, 4279, 4444, 4449, 4453, 4455, 4458, 4469, 4933, 4934, 4939, 4954, 4955, 4956, 4957, 4958, 4963, 4973, 4974, 4975, 4979, 4984, 5069, 5266, 5275, 5276, 5277, 5278, 5279, 5280, 5281, 5287, 5288, 5289, 5290, 5291, 5292, 5294, 5295, 5296, 5297, 5298, 5299, 5302, 5304, 5307, 5308, 5309, 5310, 5317, 5318, 5319, 5321, 5322, 5323, 5324, 5325, 5326, 5327, 5328, 5329, 5330, 5331, 5337, 5338, 5339, 5340, 5341, 5342, 5343, 5344, 5345, 5346, 5347, 5349, 5351, 5354, 5356, 5357, 5358, 5359, 5360, 5361, 5362, 5363, 5364, 5365, 5366, 5367, 5368, 5369, 5370, 5371, 5372, 5373, 5374, 5375, 5376, 5377, 5378, 5379, 5380, 5381, 5564, 5565, 5592, 5593, 5602, 5614, 5615, 5616, 5620, 5621, 5622, 5634, 5980, 6169, 6170, 6171, 6172, 6173, 6189, 6190, 6191, 6192, 6193, 6194, 6195, 6196, 6206, 6285, 6286, 6470, 6471, 6486, 6487, 6488, 6489, 6492, 6493, 6494, 6562, 6563, 6581, 6598, 6599, 6600, 6604, 6605, 6611, 6612, 6613, 6614, 6615, 6936, 6942, 6944, 6969, 6971, 6972, 6973, 6974, 6975, 6980, 6987, 6988, 6989, 6999, 7000, 7019, 7020, 7021, 7022, 7023, 7024, 7025, 7026, 7027, 7028, 7029, 7030, 7031, 7032, 7033, 7039, 7040, 7041, 7042, 7043, 7044, 7045, 7046, 7047, 7048, 7049, 7050, 7057, 7059, 7086, 7093, 7094, 7096, 7124, 7125, 7126, 7146, 7164, 7165, 7166, 7167, 7168, 7513, 7519, 7520, 7521, 7522, 7523]\n"
     ]
    }
   ],
   "source": [
    "print('ball\\n',ball)\n",
    "print('bat\\n',bat)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 71,
   "metadata": {},
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "523 1\n524 2\n1964 6\n1999 7\n2095 8\n3400 11\n4143 14\n"
     ]
    },
    {
     "output_type": "execute_result",
     "data": {
      "text/plain": [
       "[523, 736, 1788, 1999, 2648, 3400, 3949, 4143, 5940, 7401]"
      ]
     },
     "metadata": {},
     "execution_count": 71
    }
   ],
   "source": [
    "frames_sorted = sorted(frames_final)\n",
    "# frames_sorted\n",
    "start =0 \n",
    "frames_from_int=[]\n",
    "int_sz = 0\n",
    "for j,i in enumerate(frames_sorted[:-1]):\n",
    "\n",
    "    if abs(i-frames_sorted[j+1])<120:\n",
    "        int_sz+=1\n",
    "        print(i, j+1)\n",
    "        continue\n",
    "    else: \n",
    "        frames_from_int.append(frames_sorted[start+int_sz//3])\n",
    "        # print(start)\n",
    "        start = j+1\n",
    "        int_sz = 1\n",
    "frames_from_int.append(frames_sorted[start+int_sz//3])\n",
    "frames_from_int\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 47,
   "metadata": {},
   "outputs": [
    {
     "output_type": "execute_result",
     "data": {
      "text/plain": [
       "[523,\n",
       " 1999,\n",
       " 3455,\n",
       " 4143,\n",
       " 5940,\n",
       " 7401,\n",
       " 3949,\n",
       " 4256,\n",
       " 2112,\n",
       " 3400,\n",
       " 2095,\n",
       " 2648,\n",
       " 1788,\n",
       " 1964,\n",
       " 736,\n",
       " 524,\n",
       " 568]"
      ]
     },
     "metadata": {},
     "execution_count": 47
    }
   ],
   "source": [
    "frames_final"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 24,
   "metadata": {},
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "[506, 1964, 2038, 4256, 5980, 622, 1962, 2638, 3118, 3400, 4190, 5846, 5940, 7356, 560, 562, 564, 566, 568, 570, 726, 728, 730, 736]\n[506, 1964, 2038, 4256, 5980, 622, 1962, 2638, 3118, 3400, 4190, 5846, 5940, 7356, 560, 562, 564, 566, 568, 570, 726, 728, 730, 736, 528, 726, 884, 1222, 1306, 1766, 1982, 2044, 2404, 2516, 2764, 2892, 3438, 3542, 3806, 3944, 4200, 4444, 4934, 5266, 5564, 6170, 6286, 6470, 6562, 6936, 7520]\n"
     ]
    }
   ],
   "source": [
    "# new_out =[]\n",
    "print(new_out)\n",
    "for j,i in enumerate(bat):\n",
    "    if j==0:\n",
    "        continue\n",
    "    if i - bat[j-1]>50:\n",
    "        new_out.append(i)\n",
    "\n",
    "print(new_out)\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 39,
   "metadata": {},
   "outputs": [
    {
     "output_type": "execute_result",
     "data": {
      "text/plain": [
       "[1058, 793, 1163, 969]"
      ]
     },
     "metadata": {},
     "execution_count": 39
    }
   ],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 63,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "496 528 757.9406309203907\nsaving frame  496\n497 528 757.9406309203907\n498 528 755.8438992278762\ndeleted  496\nsaving frame  498\n499 528 723.3429338840602\ndeleted  498\nsaving frame  499\n500 528 707.6510439475095\ndeleted  499\nsaving frame  500\n501 528 679.3651448227234\ndeleted  500\nsaving frame  501\n502 528 659.2738429514703\ndeleted  501\nsaving frame  502\n503 528 659.2738429514703\n504 528 639.5349873150021\ndeleted  502\nsaving frame  504\n505 528 627.0135564722664\ndeleted  504\nsaving frame  505\n508 528 577.3958780594126\ndeleted  505\nsaving frame  508\n513 528 554.0730998704051\ndeleted  508\nsaving frame  513\n514 528 549.765404513598\ndeleted  513\nsaving frame  514\n515 528 548.8642819495544\ndeleted  514\nsaving frame  515\n516 528 543.2531638195952\ndeleted  515\nsaving frame  516\n517 528 550.3180898353243\n518 528 549.3641779366397\n519 528 560.4712303053566\n523 528 496.23482344551354\ndeleted  516\nsaving frame  523\n524 528 496.23482344551354\n529 528 759.582121959173\n529 560 2433.135425741855\n530 560 2433.135425741855\n533 560 2819.2454664324637\n1962 1964 1209.0334982952293\n1963 1964 1209.0334982952293\n1964 1964 875.0\nsaving frame  1964\n1968 1964 1206.0335816220045\n1968 1981 1280.2956689765065\n1969 1981 1278.3102127418056\n1971 1981 1275.4532527693832\n1992 1981 1759.4058656262346\n1992 1982 1759.4058656262346\n1992 1983 1734.2341249093215\n1992 1984 1723.0142193261204\n1992 1985 1664.8867829375065\n1992 1986 1730.8183035778193\n1992 2035 573.464035489585\ndeleted  1964\nsaving frame  1992\n1993 2035 572.7634764892049\ndeleted  1992\nsaving frame  1993\n1994 2035 572.7634764892049\n1997 2035 563.7242233574854\ndeleted  1993\nsaving frame  1997\n1998 2035 560.9322597248263\ndeleted  1997\nsaving frame  1998\n1999 2035 430.4056226398535\ndeleted  1998\nsaving frame  1999\n2000 2035 430.4056226398535\n2004 2035 549.2412948786717\n2007 2035 541.7130236573605\n2014 2035 536.8090908321133\n2015 2035 536.8090908321133\n2016 2035 536.8090908321133\n2017 2035 536.8090908321133\n2018 2035 536.8090908321133\n2019 2035 536.8090908321133\n2020 2035 536.0643618074233\n2021 2035 536.0643618074233\n2022 2035 536.0643618074233\n2023 2035 536.0643618074233\n2026 2035 536.0643618074233\n2027 2035 536.0643618074233\n2028 2035 536.0643618074233\n2030 2035 535.3204647685347\n2031 2035 534.5774031887244\n2032 2035 534.5774031887244\n2033 2035 531.1609925436919\n2034 2035 531.8279797077247\n2035 2035 529.0831692654757\n2036 2035 528.3370136570028\n2036 2043 1130.2433366315415\n3399 3419 704.9886523909445\nsaving frame  3399\n3400 3419 704.9886523909445\n3455 3419 862.2000927858916\n3455 3438 738.9215113934632\n3455 3439 739.5363412300981\n3455 3444 939.5030601333877\n3455 3445 938.1822850597852\n3455 3449 250.5673562138532\ndeleted  3399\nsaving frame  3455\n3455 3455 821.6355639819883\n4143 4183 286.93727537564723\nsaving frame  4143\n4190 4183 1105.349266069327\n4190 4197 930.5847623940551\n4191 4197 894.8659117432063\n4192 4197 857.7575414999276\n4193 4197 857.7575414999276\n4232 4197 485.7417009069738\n4232 4200 519.754749858046\n4232 4201 538.9666037891402\n4232 4202 591.2512156435706\n4232 4203 593.9966329870903\n4232 4204 650.30223742503\n4232 4205 651.0360973095117\n4232 4206 675.2666140125691\n4232 4207 723.486696491373\n4232 4208 758.0534281961925\n4232 4209 803.4880210681426\n4232 4210 833.7079824494905\n4232 4211 833.7079824494905\n4232 4212 882.5383844343543\n4232 4213 924.8053849324192\n4232 4214 978.694027773747\n4232 4215 1014.7418390901205\n4232 4216 1069.4227414825252\n4232 4217 1069.8560650853926\n4232 4218 1136.5795176757322\n4232 4219 1201.6193240789696\n4232 4220 1288.9259094300185\n4232 4221 1397.950285239071\n4232 4222 1445.838856857845\n4232 4223 1446.5102834062398\n4232 4255 1210.784869413225\n4235 4255 1121.3964508593738\n4237 4255 1043.472088749862\n4241 4255 836.6367192515519\n4242 4255 776.9530230329244\n4243 4255 645.962847228848\n4244 4255 576.3271640309869\n4245 4255 511.53103522660285\n4252 4255 362.93250061133955\n4256 4255 525.8631000555182\n4256 4256 742.3880386967452\n5940 5980 813.5699109480389\nsaving frame  5940\n5941 5980 813.5699109480389\n5950 5980 1105.3994753029333\n5979 5980 1042.5416058843887\n5980 5980 1019.2453090399779\n"
     ]
    },
    {
     "output_type": "execute_result",
     "data": {
      "text/plain": [
       "[523, 1999, 3455, 4143, 5940]"
      ]
     },
     "metadata": {},
     "execution_count": 63
    }
   ],
   "source": [
    "import math\n",
    "i =0 \n",
    "j =0 \n",
    "th = 1000\n",
    "ldist = th\n",
    "frames =[]\n",
    "while (i<len(ball) and j< len(bat)):\n",
    "    # print(i,j)\n",
    "    # print('checking')\n",
    "    while (i<len(ball) and j < len(bat)) and abs(ball[i]- bat[j])<50:\n",
    "       \n",
    "        # print('here i am')\n",
    "        bcx,bcy,cx,cy = cor_ball[i][0],cor_ball[i][1], cor_bat[j][0]+cor_bat[j][2]//2,\\\n",
    "            cor_bat[j][1]+cor_bat[j][3]//2,\n",
    "        dist = math.sqrt(((bcx-cx)*(bcx-cx))+((bcy-cy)*(bcy-cy)))\n",
    "        print(ball[i],bat[j], dist)\n",
    "        if dist<ldist:\n",
    "            if ldist !=th:\n",
    "                print('deleted ', frames[-1])\n",
    "                del frames[-1]\n",
    "                # del frames[-1]\n",
    "\n",
    "                # del frames[-1]\n",
    "            print('saving frame ' ,ball[i])\n",
    "            frames.append(ball[i])\n",
    "            # frames.append(bat[i])\n",
    "            # frames.append(bat[j])\n",
    "            ldist = dist\n",
    "        if ball[i]>bat[j]:\n",
    "            j+=1\n",
    "            # ldist=th\n",
    "        else:\n",
    "            i+=1\n",
    "            # ldist = th\n",
    "            \n",
    "    try:\n",
    "        if ball[i]>bat[j]:\n",
    "            j+=1 \n",
    "            ldist =th\n",
    "        else:\n",
    "            i+=1\n",
    "            ldist = th \n",
    "    except:\n",
    "        print(i, j)\n",
    "        continue \n",
    "        \n",
    "frames"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 64,
   "metadata": {},
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "7401 7523 454.954942824012\nsaving frame  7401\n7401 7522 482.21571936219584\n7401 7521 481.5080477001397\n7401 7520 546.0375444967132\n7401 7519 546.0375444967132\n7401 7513 632.5582977085985\n5980 5980 1019.2453090399779\n4256 4279 256.11325619733157\nsaving frame  4256\n4256 4278 255.43100829773977\ndeleted  5940\nsaving frame  4256\n4256 4277 311.4498354470588\n4256 4276 379.0052770081177\n4256 4275 429.54045211132325\n4256 4274 458.4157501657202\n4256 4273 457.5259555478793\n4256 4268 719.2169352844801\n4256 4267 795.6142281281802\n4256 4266 788.2721611220328\n4256 4265 796.9868254870968\n4256 4264 796.9868254870968\n4256 4262 867.4203133429606\n4256 4261 844.0023696649198\n4256 4257 823.3595812280319\n4256 4256 742.3880386967452\n4256 4255 525.8631000555182\n4252 4255 362.93250061133955\n4252 4223 660.4362194792167\n4245 4223 851.657208036191\n4244 4223 904.8607627696098\n4243 4223 964.168035147401\n4242 4223 1077.671564067643\n4241 4223 1130.913347697338\n4237 4223 1317.331393385886\n4235 4223 1386.5446260398546\n4232 4223 1446.5102834062398\n4193 4223 880.3709445455364\n4193 4222 880.3618574200043\n4193 4221 888.9904386437461\n4193 4220 807.421203585836\n4193 4219 754.560799405853\n4193 4218 723.0006915626014\n4193 4217 698.7481663661093\n4193 4216 697.7922326882122\n4193 4215 670.2313033572813\n4193 4214 680.3888594032093\n4193 4213 662.5299993207855\n4193 4212 661.0121027636333\n4193 4211 642.870126853006\n4193 4210 642.870126853006\n4193 4209 658.4588066082798\n4193 4208 658.2780567510966\n4193 4207 673.4819967898177\n4193 4206 679.7764338368902\n4193 4205 704.4693038025149\n4193 4204 704.868782114799\n4193 4203 704.7162549565605\n4193 4202 749.7899705917652\n4193 4201 757.4655107660018\n4193 4200 790.8198277736844\n4193 4197 857.7575414999276\n4193 4183 1078.2286399460922\n4192 4183 1078.2286399460922\n4191 4183 1091.9047577513343\n4190 4183 1105.349266069327\n4143 4183 286.93727537564723\n3455 3583 560.9179975718376\nsaving frame  3455\n3455 3562 643.8540207220889\n3455 3556 634.9503917630101\n3455 3555 617.5597137119616\n3455 3551 579.0138167608783\n3455 3547 554.8279012450618\ndeleted  5940\nsaving frame  3455\n3455 3546 554.8279012450618\n3455 3545 563.952125627699\n3455 3543 609.7614287571821\n3455 3542 670.3379744576612\n3455 3539 809.8401076755831\n3455 3488 590.4726920019249\n3455 3458 930.473535357132\n3455 3455 821.6355639819883\n3455 3449 250.5673562138532\ndeleted  5940\nsaving frame  3455\n3400 3449 249.64775184247102\ndeleted  5940\nsaving frame  3400\n3400 3445 808.2487240942605\n3400 3444 809.3120535368295\n3400 3439 586.2371533773683\n3400 3438 585.4955166352685\n3400 3419 704.9886523909445\n2711 2775 1693.245404541232\n2711 2772 1645.9662815501415\n2711 2771 1569.709527269297\n2711 2770 1558.3747944573538\n2711 2769 1534.4288839825715\n2711 2768 1462.6968243624515\n2711 2767 1464.0262975780183\n2711 2766 1465.1842887500534\n2711 2765 1449.8799950340717\n2711 2764 1349.5840099823354\n2711 2576 1146.3420083029323\n2682 2576 616.2207396704529\nsaving frame  2682\n2681 2576 697.3643237218262\n2680 2576 770.7788269017254\n2679 2576 836.9593777478092\n2678 2576 885.3208457954664\n2677 2576 885.3208457954664\n2676 2576 928.3280670107955\n2675 2576 958.0240080499027\n2674 2576 979.0291109052887\n2673 2576 982.4586505293747\n2672 2576 964.2675977134148\n2671 2576 964.2675977134148\n2670 2576 938.9595305443148\n2669 2576 903.0088593142373\n2668 2576 861.7081872652714\n2667 2576 817.5383783040402\n2666 2576 766.7261571121726\n2665 2576 766.7261571121726\n2664 2576 705.7853781426759\n2663 2576 651.6763000140484\n2662 2576 624.6983271948149\n2661 2576 599.3479790572418\ndeleted  5940\nsaving frame  2661\n2660 2576 567.2971002922543\ndeleted  5940\nsaving frame  2660\n2659 2576 567.2971002922543\n2658 2576 527.8730908087663\ndeleted  5940\nsaving frame  2658\n2657 2576 504.8960289010006\ndeleted  5940\nsaving frame  2657\n2656 2576 476.18693808209395\ndeleted  5940\nsaving frame  2656\n2655 2576 454.22131169728266\ndeleted  5940\nsaving frame  2655\n2654 2576 415.7884077268148\ndeleted  5940\nsaving frame  2654\n2653 2576 415.7884077268148\n2652 2576 395.4263521820467\ndeleted  5940\nsaving frame  2652\n2651 2576 378.6251444370804\ndeleted  5940\nsaving frame  2651\n2650 2576 361.8425071768103\ndeleted  5940\nsaving frame  2650\n2649 2576 353.40911137094355\ndeleted  5940\nsaving frame  2649\n2648 2576 344.4880259167218\ndeleted  5940\nsaving frame  2648\n2647 2576 344.4880259167218\n2646 2576 346.5847659664227\n2645 2576 355.78785814021256\n2644 2576 355.78785814021256\n2643 2576 375.479693192588\n2642 2576 375.479693192588\n2641 2576 387.489354692487\n2640 2576 403.8341243629617\n2639 2576 420.9429890139519\n2638 2576 420.9429890139519\n2036 2127 762.9134944408835\nsaving frame  2036\n2036 2126 800.1781051740918\n2036 2125 816.7214947581581\n2036 2124 815.7236051506662\n2036 2123 814.6686442965631\n2036 2122 826.4090996594846\n2036 2121 842.0593803289647\n2036 2120 859.0145516811691\n2036 2119 878.0142367866252\n2036 2118 899.0200220239814\n2036 2117 907.2915738614572\n2036 2116 907.2667744384779\n2036 2115 917.4905994068822\n2036 2114 911.8464783065184\n2036 2113 924.4311764539316\n2036 2112 930.888822577648\n2036 2111 941.6039507138869\n2036 2110 941.541289588513\n2036 2109 941.541289588513\n2036 2108 942.7963725004462\n2036 2107 942.6690829766297\n2036 2106 942.3274377837038\n2036 2105 946.3920963321704\n2036 2104 932.0354070527578\n2036 2103 932.0354070527578\n2036 2102 935.4635214694371\n2036 2100 926.0653324685036\n2036 2099 921.0043430950801\n2036 2098 906.4132611563006\n2036 2097 898.8398077521935\n2036 2096 898.6534370935216\n2036 2095 899.8855482782241\n2036 2094 898.8937645795526\n2036 2093 887.7567234327206\n2036 2092 874.1224170561009\n2036 2091 858.6093407365191\n2036 2090 858.8113879077291\n2036 2089 839.9077330278607\n2036 2088 833.3744656515461\n2036 2087 829.0006031360894\n2036 2086 827.5004531720814\n2036 2085 827.488972736193\n2036 2084 827.2683965920613\n2036 2083 824.4865068635144\n2036 2082 825.3496228871738\n2036 2081 823.4087684740794\n2036 2079 816.6694557775502\n2036 2078 816.6694557775502\n2036 2076 822.5746166762016\n2036 2075 822.5746166762016\n2036 2074 821.2484398767525\n2036 2073 825.8407836865409\n2036 2072 830.4143544038723\n2036 2071 831.6850365372699\n2036 2070 836.1530960296685\n2036 2069 836.1530960296685\n2036 2068 839.3545138974354\n2036 2067 825.7027310115911\n2036 2066 825.360527284895\n2036 2065 835.1580688707977\n2036 2061 841.8004514135164\n2036 2059 859.3421902827768\n2036 2058 863.173794782951\n2036 2057 863.173794782951\n2036 2055 900.8940004240233\n2036 2054 915.2977657571332\n2036 2053 947.9266849287449\n2036 2052 956.7026706349261\n2036 2051 956.7026706349261\n2036 2049 1011.8596740655297\n2036 2048 1025.5032910722423\n2036 2047 1044.4831257612543\n2036 2046 1043.579417198327\n2036 2045 1072.9930102288645\n2036 2044 1083.184656464446\n2036 2043 1130.2433366315415\n2036 2035 528.3370136570028\ndeleted  5940\nsaving frame  2036\n2035 2035 529.0831692654757\n2035 1986 1676.8404813815773\n2034 1986 1679.94404668727\n2033 1986 1678.952351914729\n2032 1986 1683.0478305740453\n2031 1986 1683.0478305740453\n2030 1986 1683.1770554519808\n2028 1986 1683.3068644783696\n2027 1986 1683.3068644783696\n2026 1986 1683.3068644783696\n2023 1986 1683.3068644783696\n2022 1986 1683.3068644783696\n2021 1986 1683.3068644783696\n2020 1986 1683.3068644783696\n2019 1986 1683.4372575180817\n2018 1986 1683.4372575180817\n2017 1986 1683.4372575180817\n2016 1986 1683.4372575180817\n2015 1986 1683.4372575180817\n2014 1986 1683.4372575180817\n2007 1986 1687.7967887159875\n2004 1986 1696.9823216521731\n2000 1986 1192.7992287053173\n1999 1986 1192.7992287053173\n1998 1986 1712.1150662265663\n1997 1986 1715.2212685248512\n1994 1986 1728.9664542726096\n1993 1986 1728.9664542726096\n1992 1986 1730.8183035778193\n1971 1986 1248.7121365631072\n1971 1985 1178.6746794599433\n1971 1984 1236.0582510545366\n1971 1983 1248.4686620015739\n1971 1982 1275.4532527693832\n1971 1981 1275.4532527693832\n1971 1964 1201.0416312518064\n1969 1964 1204.0336374038725\n1968 1964 1206.0335816220045\n1964 1964 875.0\n1964 1816 414.5672442439224\ndeleted  5940\nsaving frame  1964\n1963 1816 911.6605728010837\n1962 1816 911.6605728010837\n632 746 1196.7489293916246\n632 736 782.7036476214992\nsaving frame  632\n632 731 838.098442905128\n632 730 837.5452226596484\n632 729 837.8812565035693\n632 728 846.0543717752423\n632 727 853.8717702325098\n632 726 865.520074868284\n632 725 890.1247103636658\n632 571 2049.693879582998\n631 571 2049.693879582998\n622 571 1790.5753265361388\n621 571 1761.3460761587996\n533 571 2667.9109805239004\n533 570 2638.9399386875025\n533 569 2638.598302129371\n533 568 2602.786967848118\n533 567 2585.8029700655848\n533 566 2587.069577726892\n533 565 2600.0807679762565\n533 564 2627.583871163773\n533 563 2699.4406828081997\n533 562 2762.4838461066156\n533 561 2819.2454664324637\n533 560 2819.2454664324637\n533 528 1155.5725853445988\n530 528 759.582121959173\ndeleted  5940\nsaving frame  530\n529 528 759.582121959173\n524 528 496.23482344551354\ndeleted  5940\nsaving frame  524\n"
     ]
    },
    {
     "output_type": "execute_result",
     "data": {
      "text/plain": [
       "[7401, 3949, 4256, 2112, 3400, 2095, 2648, 1788, 1964, 736, 524, 568]"
      ]
     },
     "metadata": {},
     "execution_count": 64
    }
   ],
   "source": [
    "import math\n",
    "i =len(ball)-1 \n",
    "j =len(bat)-1\n",
    "th = 1000\n",
    "ldist = th\n",
    "frames_rev =[]\n",
    "while (i>=0 and j>=0):\n",
    "    # print(i,j)\n",
    "    # print('checking')\n",
    "    while (i>=0 and j>=0) and abs(ball[i]- bat[j])<150:\n",
    "       \n",
    "        # print('here i am')\n",
    "        bcx,bcy,cx,cy = cor_ball[i][0],cor_ball[i][1], cor_bat[j][0]+cor_bat[j][2]//2,\\\n",
    "            cor_bat[j][1]+cor_bat[j][3]//2,\n",
    "        dist = math.sqrt(((bcx-cx)*(bcx-cx))+((bcy-cy)*(bcy-cy)))\n",
    "        print(ball[i],bat[j], dist)\n",
    "        if dist<ldist:\n",
    "            if ldist !=th:\n",
    "                print('deleted ', frames[-1])\n",
    "                del frames_rev[-1]\n",
    "                del frames_rev[-1]\n",
    "\n",
    "                # del frames[-1]\n",
    "            print('saving frame ' ,ball[i])\n",
    "            frames_rev.append(ball[i])\n",
    "            frames_rev.append(bat[i])\n",
    "            # frames.append(bat[j])\n",
    "            ldist = dist\n",
    "        if ball[i]>bat[j]:\n",
    "            i-=1\n",
    "            # ldist=th\n",
    "        else:\n",
    "            j-=1\n",
    "            # ldist = th\n",
    "            \n",
    "    try:\n",
    "        if ball[i]>bat[j]:\n",
    "            i-=1 \n",
    "            ldist =th\n",
    "        else:\n",
    "            j-=1\n",
    "            ldist = th \n",
    "    except:\n",
    "        print(i, j)\n",
    "        continue \n",
    "frames_rev"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 65,
   "metadata": {},
   "outputs": [
    {
     "output_type": "execute_result",
     "data": {
      "text/plain": [
       "[523,\n",
       " 1999,\n",
       " 3455,\n",
       " 4143,\n",
       " 5940,\n",
       " 7401,\n",
       " 3949,\n",
       " 4256,\n",
       " 2112,\n",
       " 3400,\n",
       " 2095,\n",
       " 2648,\n",
       " 1788,\n",
       " 1964,\n",
       " 736,\n",
       " 524,\n",
       " 568]"
      ]
     },
     "metadata": {},
     "execution_count": 65
    }
   ],
   "source": [
    "frames_final = frames+frames_rev\n",
    "frames_final"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 231,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "reading fram 200\n",
      "reading fram 400\n",
      "reading fram 600\n",
      "reading fram 800\n",
      "reading fram 1000\n",
      "reading fram 1200\n",
      "reading fram 1400\n",
      "reading fram 1600\n",
      "reading fram 1800\n",
      "reading fram 2000\n",
      "reading fram 2200\n",
      "reading fram 2400\n",
      "reading fram 2600\n",
      "reading fram 2800\n",
      "reading fram 3000\n",
      "reading fram 3200\n",
      "reading fram 3400\n",
      "reading fram 3600\n",
      "reading fram 3800\n",
      "reading fram 4000\n",
      "reading fram 4200\n",
      "reading fram 4400\n",
      "reading fram 4600\n",
      "reading fram 4800\n",
      "reading fram 5000\n",
      "reading fram 5200\n",
      "reading fram 5400\n",
      "reading fram 5600\n",
      "reading fram 5800\n",
      "reading fram 6000\n",
      "reading fram 6200\n",
      "reading fram 6400\n",
      "reading fram 6600\n",
      "reading fram 6800\n",
      "reading fram 7000\n",
      "reading fram 7200\n",
      "reading fram 7400\n",
      "reading fram 7600\n"
     ]
    }
   ],
   "source": [
    "import cv2\n",
    "import numpy as np\n",
    "import tqdm\n",
    "import shutil\n",
    "# new_out = [510, 1965, 1995, 2010, 2035, 2040, 3455, 4260, 5980]\n",
    "name = 'results/centernet/'\n",
    "\n",
    "try:\n",
    "    shutil.rmtree(name)\n",
    "except:\n",
    "    print('does not exist')\n",
    "os.makedirs(name, exist_ok=True)\n",
    "cap = cv2.VideoCapture('../data/video.mp4')\n",
    " \n",
    "# fourcc = cv2.VideoWriter_fourcc(*'XVID')\n",
    "# out = cv2.VideoWriter('results/red_cricket.mp4',fourcc, 20, (1280,720))\n",
    "k = 0\n",
    "# frames_int = [ frames_sorted[i] for i in frames_from_int]\n",
    "while cap.isOpened():\n",
    "    \n",
    "    ret, frame = cap.read()\n",
    "    # print(frame.shape)\n",
    "    # break\n",
    "    if ret == True:\n",
    "        if k in frames:\n",
    "            cv2.imwrite(name+str(k)+'.jpg',frame)\n",
    "     \n",
    "        # b = cv2.resize(frame,(1280,720),fx=0,fy=0, interpolation = cv2.INTER_CUBIC)\n",
    "        # out.write(b)\n",
    "    else:\n",
    "        break\n",
    "    k+=1\n",
    "    if k%200 ==0:\n",
    "        print(f'reading fram {k}')\n",
    "    \n",
    "cap.release()\n",
    "# out.release()\n",
    "cv2.destroyAllWindows()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "1999\n"
     ]
    }
   ],
   "source": [
    "import cv2\n",
    "\n",
    "cap = cv2.VideoCapture(\"data/red_cricket.avi\")\n",
    "length = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))\n",
    "print( length )"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "output_type": "execute_result",
     "data": {
      "text/plain": [
       "[240,\n",
       " [{'name': 'baseball bat',\n",
       "   'percentage_probability': 51.07940435409546,\n",
       "   'box_points': [1122, 458, 1198, 583]}]]"
      ]
     },
     "metadata": {},
     "execution_count": 4
    }
   ],
   "source": [
    "content[1]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "            # break\n",
    "            \n",
    "        \n",
    "            \n",
    "        # break"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "output_type": "execute_result",
     "data": {
      "text/plain": [
       "[506,\n",
       " 510,\n",
       " 512,\n",
       " 1964,\n",
       " 1966,\n",
       " 1996,\n",
       " 2002,\n",
       " 2010,\n",
       " 2038,\n",
       " 2040,\n",
       " 2042,\n",
       " 4256,\n",
       " 4258,\n",
       " 4260,\n",
       " 5980]"
      ]
     },
     "metadata": {},
     "execution_count": 6
    }
   ],
   "source": [
    "new_out"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {},
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "(1080, 2280, 3)\n"
     ]
    },
    {
     "output_type": "execute_result",
     "data": {
      "text/plain": [
       "(500, 1000, 3)"
      ]
     },
     "metadata": {},
     "execution_count": 13
    }
   ],
   "source": [
    "import cv2 \n",
    "img = cv2.imread('results.jpg')\n",
    "print(img.shape)\n",
    "img = cv2.resize(img, (1000,500))\n",
    "img.shape "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {},
   "outputs": [
    {
     "output_type": "execute_result",
     "data": {
      "text/plain": [
       "True"
      ]
     },
     "metadata": {},
     "execution_count": 15
    }
   ],
   "source": [
    "cv2.imshow('',img)\n",
    "cv2.waitKey(0)\n",
    "cv2.destroyAllWindows()\n",
    "cv2.imwrite('results1.jpg', img)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "348\n"
     ]
    }
   ],
   "source": [
    "for r,d, files in os.walk('results/frames'):\n",
    "    print(len(files))\n",
    "with open('files.txt','w') as f:\n",
    "    f.write('\\n'.join([i for i in files]))\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "Frame rate : 60.0\n"
     ]
    }
   ],
   "source": [
    "import cv2\n",
    "\n",
    "video = cv2.VideoCapture(\"results/detfps60fd1normal.avi\");\n",
    "video.set(cv2.CAP_PROP_FPS, int(20))\n",
    "\n",
    "fourcc = cv2.VideoWriter_fourcc(*'XVID')\n",
    "out = cv2.VideoWriter('results/detfps20fd1normal.mp4',fourcc, 20, (1280,720))\n",
    "\n",
    "if __name__ == '__main__':\n",
    "    print(\"Frame rate : {0}\".format(video.get(cv2.CAP_PROP_FPS)))\n",
    "    while video.isOpened():\n",
    "        ret1, frame2 = video.read()\n",
    "        # cv2.imshow(\"Changed\", frame2)\n",
    "\n",
    "        \n",
    "        if ret1:\n",
    "            b = cv2.resize(frame2,(1280,720),fx=0,fy=0, interpolation = cv2.INTER_CUBIC)\n",
    "            cv2.imshow(\"Changed\", b)\n",
    "            if cv2.waitKey(30) & 0xFF == ord('q'):  # press q to quit\n",
    "                break\n",
    "            out.write(b)\n",
    "        else:\n",
    "            break\n",
    "\n",
    "video.release()\n",
    "out.release()\n",
    "cv2.destroyAllWindows()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {},
   "outputs": [
    {
     "output_type": "execute_result",
     "data": {
      "text/plain": [
       "    1    2    3    4    5    6    7    8    9    10   11\n",
       "0                                                       \n",
       "0  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN\n",
       "1  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN\n",
       "2  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN\n",
       "3  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN\n",
       "4  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN  NaN"
      ],
      "text/html": "<div>\n<style scoped>\n    .dataframe tbody tr th:only-of-type {\n        vertical-align: middle;\n    }\n\n    .dataframe tbody tr th {\n        vertical-align: top;\n    }\n\n    .dataframe thead th {\n        text-align: right;\n    }\n</style>\n<table border=\"1\" class=\"dataframe\">\n  <thead>\n    <tr style=\"text-align: right;\">\n      <th></th>\n      <th>1</th>\n      <th>2</th>\n      <th>3</th>\n      <th>4</th>\n      <th>5</th>\n      <th>6</th>\n      <th>7</th>\n      <th>8</th>\n      <th>9</th>\n      <th>10</th>\n      <th>11</th>\n    </tr>\n    <tr>\n      <th>0</th>\n      <th></th>\n      <th></th>\n      <th></th>\n      <th></th>\n      <th></th>\n      <th></th>\n      <th></th>\n      <th></th>\n      <th></th>\n      <th></th>\n      <th></th>\n    </tr>\n  </thead>\n  <tbody>\n    <tr>\n      <th>0</th>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n    </tr>\n    <tr>\n      <th>1</th>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n    </tr>\n    <tr>\n      <th>2</th>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n    </tr>\n    <tr>\n      <th>3</th>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n    </tr>\n    <tr>\n      <th>4</th>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n      <td>NaN</td>\n    </tr>\n  </tbody>\n</table>\n</div>"
     },
     "metadata": {},
     "execution_count": 19
    }
   ],
   "source": [
    "import pandas as pd \n",
    "data = pd.read_csv('results/tf_output.csv', header=None, index_col=[0], skiprows=[0])\n",
    "data.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 207,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "\"\n\n\"\n\n"
     ]
    }
   ],
   "source": [
    "with open('results/tf_output.csv', 'r') as f:\n",
    "    content = f.readlines()\n",
    "out =[]\n",
    "for i in content:\n",
    "    if '[' in i:\n",
    "        arr1=[]\n",
    "        i = i.replace(']',',')\n",
    "        i = i.replace('[',',')\n",
    "        for j in i.split(',')[1:]:\n",
    "            # print(j, len(j))\n",
    "            if len(j)>1:\n",
    "                if '[' in j:\n",
    "                    j = j.replace('[',',')\n",
    "                if ']' in j:\n",
    "                    j = j.replace(']', ',')\n",
    "                try:\n",
    "                    arr1.append(int(j))\n",
    "                except:\n",
    "                    print(j)\n",
    "        classes =[]\n",
    "        for k in range(1,len(arr1), 6):\n",
    "            # print(k)\n",
    "            if k ==0:\n",
    "                continue\n",
    "            classes.append(arr1[k-3])\n",
    "        out.append([arr1[-1],classes, arr1])\n",
    "        # print(arr1)\n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 229,
   "metadata": {
    "tags": []
   },
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "222 62.24146527838174\n243 430.4880950734875\ndeleting frame  243 430.4880950734875\n244 430.4880950734875\n701 262.1907702418222\ndeleting frame  701 261.70594185077266\n702 261.70594185077266\ndeleting frame  702 73.00684899377592\n703 73.00684899377592\ndeleting frame  703 65.30696746902278\n704 65.30696746902278\n836 428.9533774199709\ndeleting frame  836 422.33872661644466\n842 422.33872661644466\n1321 221.05655384991417\ndeleting frame  1321 210.28789789238942\n1322 210.28789789238942\ndeleting frame  1322 106.30145812734649\n1362 106.30145812734649\n1644 258.81460546112925\n1764 2.0\n2127 392.9796432386797\n2889 315.9050490258109\n3145 497.92569726817675\ndeleting frame  3145 462.9222396904258\n3146 462.9222396904258\ndeleting frame  3146 422.30320860727545\n3147 422.30320860727545\n3956 194.28072472584614\n4212 401.0486254807514\n4460 352.7846368537043\ndeleting frame  4460 255.49363984256047\n4466 255.49363984256047\n4975 386.8087382673768\n5591 336.40005945302687\n5956 293.58644382872995\n6465 85.37564055396598\n6628 399.9299938739279\n7010 391.1278563334501\ndeleting frame  7010 391.10356684643006\n7011 391.10356684643006\n7130 208.6935552430884\n"
     ]
    }
   ],
   "source": [
    "# f= open('results/target_frames.txt', 'w')\n",
    "import math\n",
    "frames =[]\n",
    "th = 500\n",
    "ldst =th\n",
    "for i in out:\n",
    "    if len(set(i[1]))>1 and len(i[1])==2:\n",
    "        # f.write(str(i[0])+'\\n')\n",
    "        # print(i)\n",
    "        x1, y1,x2,y2 = i[2][0],i[2][1],i[2][6],i[2][7]\n",
    "        # print(x1, y1,x2,y2 )\n",
    "        dst = math.sqrt((x1-x2)*(x1-x2)+(y1-y2)*(y1-y2))\n",
    "        if dst<500:\n",
    "            if len(frames)>1:\n",
    "                if i[0]-frames[-1]<50:\n",
    "                    if dst<ldst:\n",
    "                        ldst = dst\n",
    "                        print('deleting frame ',frames[-1],ldst )\n",
    "                        del frames[-1]\n",
    "                        frames.append(i[0])\n",
    "                        print(i[0],dst)\n",
    "                else:\n",
    "                    ldst = dst\n",
    "                    frames.append(i[0])\n",
    "                    print(i[0],dst)\n",
    "            else:\n",
    "                frames.append(i[0])\n",
    "                print(i[0],dst)\n",
    "        # print()\n",
    "# f.close()\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 230,
   "metadata": {},
   "outputs": [
    {
     "output_type": "execute_result",
     "data": {
      "text/plain": [
       "20"
      ]
     },
     "metadata": {},
     "execution_count": 230
    }
   ],
   "source": [
    "# import json\n",
    "# with open('results/tf_out.json', 'w') as f :\n",
    "#     json.dump(out, f,indent=2)\n",
    "\n",
    "len(frames)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 123,
   "metadata": {},
   "outputs": [
    {
     "output_type": "stream",
     "name": "stdout",
     "text": [
      "[['1035,', '645,', '1188,', '1009,', '39,', '32']] 2\n[['1035,', '645,', '1189,', '1010,', '39,', '33']] 2\n[['1032,', '712,', '1151,', '1010,', '39,', '34']] 2\n[['1013,', '732,', '1144,', '1020,', '39,', '35']] 2\n[['1119,', '633,', '1243,', '947,', '39,', '36']] 2\n[['1691,', '625,', '1706,', '636,', '37,', '59']] 2\n[['1084,', '828,', '1095,', '839,', '37,', '63]\",\"[1260,', '923,', '1278,', '939,', '37,', '63']] 2\n[['1090,', '818,', '1100,', '828,', '37,', '64']] 2\n[['1090,', '818,', '1100,', '828,', '37,', '65']] 2\n[['1669,', '592,', '1688,', '602,', '37,', '105']] 2\n"
     ]
    }
   ],
   "source": [
    "for i in val[:10]:\n",
    "    print(i[1], len(i))\n",
    "    if len(i[1])>1:\n",
    "        print(i)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ]
}