utils.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "ExecuteTime": {
     "end_time": "2020-09-12T14:12:46.572492Z",
     "start_time": "2020-09-12T14:12:43.550231Z"
    }
   },
   "outputs": [],
   "source": [
    "#import cv2\n",
    "import cv2 as cv\n",
    "import glob\n",
    "import os\n",
    "import numpy as np\n",
    "from moviepy.editor import *"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {
    "ExecuteTime": {
     "end_time": "2020-09-12T14:15:32.579695Z",
     "start_time": "2020-09-12T14:15:22.602354Z"
    }
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "the video length of Arson001_x264.mp4 is: 4187\n",
      "the video length of Arson002_x264.mp4 is: 4439\n",
      "the video length of Arson003_x264.mp4 is: 4013\n",
      "the video length of Arson005_x264.mp4 is: 1174\n",
      "the video length of Arson006_x264.mp4 is: 2815\n"
     ]
    }
   ],
   "source": [
    "#this fn seperate video to frames, set numOfFrame or sample_rate to None\n",
    "def video2frame(video_path,frame_path,numOfFrame,sample_rate):\n",
    "    file_list = os.listdir(video_path)\n",
    "    i = 0\n",
    "    for filename in file_list:\n",
    "        if i == 10:\n",
    "            break\n",
    "        video = cv.VideoCapture(video_path + str(filename))\n",
    "        \n",
    "        os.mkdir(os.path.join(frame_path,os.path.join(\"positive\"+str(i))))\n",
    "        video_length = video.get(7)\n",
    "        if video_length > 5000:\n",
    "            continue\n",
    "        if sample_rate == None:\n",
    "            sample_rate = (video_length // numOfFrame)\n",
    "            if sample_rate == 0:\n",
    "                sample_rate = 1\n",
    "        if numOfFrame == None:\n",
    "            if video_length <= 700:\n",
    "                sample_rate = int(sample_rate // 2)\n",
    "            numOfFrame = int((video_length // sample_rate))\n",
    "        print(\"the video length of %s is: %d\"%(str(filename),video_length))\n",
    "        \n",
    "        k = 0\n",
    "        while True:\n",
    "            ret,image = video.read()\n",
    "            if image is None:\n",
    "                break\n",
    "            if k % sample_rate == 0:\n",
    "                image_name = frame_path + \"positive\" + str(i) + '/' + str(int(k//sample_rate)) + '.jpg'\n",
    "                cv.imwrite(image_name,image)\n",
    "            k += 1\n",
    "            if (k / numOfFrame) == sample_rate:\n",
    "                break\n",
    "    \n",
    "        i += 2\n",
    "if __name__ == \"__main__\":\n",
    "    video_path = \"C:/Users/jingy/Jupyter/Feature/frames/Arson/Arson/\"\n",
    "    frame_path = \"C:/Users/jingy/Jupyter/Fight/frames/Arson/\"\n",
    "    video2frame(video_path,frame_path,None,5)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "the video length of Normal_Videos001_x264.mp4 is: 544\n",
      "the video length of Normal_Videos002_x264.mp4 is: 1663\n",
      "the video length of Normal_Videos004_x264.mp4 is: 917\n",
      "the video length of Normal_Videos005_x264.mp4 is: 386\n",
      "the video length of Normal_Videos007_x264.mp4 is: 985\n",
      "the video length of Normal_Videos008_x264.mp4 is: 2567\n",
      "the video length of Normal_Videos009_x264.mp4 is: 740\n",
      "the video length of Normal_Videos011_x264.mp4 is: 897\n",
      "the video length of Normal_Videos012_x264.mp4 is: 2340\n",
      "the video length of Normal_Videos013_x264.mp4 is: 1212\n",
      "the video length of Normal_Videos016_x264.mp4 is: 3631\n",
      "the video length of Normal_Videos017_x264.mp4 is: 853\n",
      "the video length of Normal_Videos020_x264.mp4 is: 485\n",
      "the video length of Normal_Videos021_x264.mp4 is: 1963\n",
      "the video length of Normal_Videos022_x264.mp4 is: 385\n",
      "the video length of Normal_Videos023_x264.mp4 is: 1769\n",
      "the video length of Normal_Videos026_x264.mp4 is: 2733\n",
      "the video length of Normal_Videos029_x264.mp4 is: 869\n",
      "the video length of Normal_Videos030_x264.mp4 is: 4427\n",
      "the video length of Normal_Videos031_x264.mp4 is: 3333\n",
      "the video length of Normal_Videos032_x264.mp4 is: 835\n",
      "the video length of Normal_Videos036_x264.mp4 is: 1323\n",
      "the video length of Normal_Videos037_x264.mp4 is: 449\n",
      "the video length of Normal_Videos038_x264.mp4 is: 2971\n",
      "the video length of Normal_Videos039_x264.mp4 is: 1792\n",
      "the video length of Normal_Videos040_x264.mp4 is: 956\n",
      "the video length of Normal_Videos043_x264.mp4 is: 1732\n",
      "the video length of Normal_Videos044_x264.mp4 is: 2511\n",
      "the video length of Normal_Videos045_x264.mp4 is: 1548\n",
      "the video length of Normal_Videos052_x264.mp4 is: 320\n",
      "the video length of Normal_Videos053_x264.mp4 is: 399\n",
      "the video length of Normal_Videos054_x264.mp4 is: 1976\n",
      "the video length of Normal_Videos055_x264.mp4 is: 251\n",
      "the video length of Normal_Videos057_x264.mp4 is: 3785\n",
      "the video length of Normal_Videos058_x264.mp4 is: 989\n",
      "the video length of Normal_Videos060_x264.mp4 is: 3072\n",
      "the video length of Normal_Videos062_x264.mp4 is: 1573\n",
      "the video length of Normal_Videos064_x264.mp4 is: 2334\n",
      "the video length of Normal_Videos065_x264.mp4 is: 866\n",
      "the video length of Normal_Videos066_x264.mp4 is: 1008\n",
      "the video length of Normal_Videos068_x264.mp4 is: 1882\n",
      "the video length of Normal_Videos069_x264.mp4 is: 923\n",
      "the video length of Normal_Videos071_x264.mp4 is: 871\n",
      "the video length of Normal_Videos072_x264.mp4 is: 952\n",
      "the video length of Normal_Videos073_x264.mp4 is: 1628\n",
      "the video length of Normal_Videos074_x264.mp4 is: 2040\n",
      "the video length of Normal_Videos075_x264.mp4 is: 3397\n",
      "the video length of Normal_Videos076_x264.mp4 is: 3072\n",
      "the video length of Normal_Videos077_x264.mp4 is: 2806\n",
      "the video length of Normal_Videos078_x264.mp4 is: 2134\n",
      "the video length of Normal_Videos079_x264.mp4 is: 2001\n",
      "the video length of Normal_Videos080_x264.mp4 is: 2344\n",
      "the video length of Normal_Videos081_x264.mp4 is: 1623\n",
      "the video length of Normal_Videos082_x264.mp4 is: 1198\n",
      "the video length of Normal_Videos083_x264.mp4 is: 586\n",
      "the video length of Normal_Videos086_x264.mp4 is: 3391\n",
      "the video length of Normal_Videos088_x264.mp4 is: 2477\n",
      "the video length of Normal_Videos092_x264.mp4 is: 1800\n",
      "the video length of Normal_Videos093_x264.mp4 is: 1740\n",
      "the video length of Normal_Videos094_x264.mp4 is: 1740\n",
      "the video length of Normal_Videos095_x264.mp4 is: 1047\n",
      "the video length of Normal_Videos096_x264.mp4 is: 779\n",
      "the video length of Normal_Videos097_x264.mp4 is: 4746\n",
      "the video length of Normal_Videos098_x264.mp4 is: 3353\n",
      "the video length of Normal_Videos099_x264.mp4 is: 2075\n",
      "the video length of Normal_Videos101_x264.mp4 is: 1538\n",
      "the video length of Normal_Videos102_x264.mp4 is: 1029\n",
      "the video length of Normal_Videos103_x264.mp4 is: 742\n",
      "the video length of Normal_Videos104_x264.mp4 is: 689\n",
      "the video length of Normal_Videos105_x264.mp4 is: 716\n",
      "the video length of Normal_Videos106_x264.mp4 is: 1944\n",
      "the video length of Normal_Videos107_x264.mp4 is: 3601\n",
      "the video length of Normal_Videos108_x264.mp4 is: 1772\n",
      "the video length of Normal_Videos109_x264.mp4 is: 1722\n",
      "the video length of Normal_Videos110_x264.mp4 is: 3132\n",
      "the video length of Normal_Videos112_x264.mp4 is: 816\n",
      "the video length of Normal_Videos113_x264.mp4 is: 694\n",
      "the video length of Normal_Videos114_x264.mp4 is: 1787\n",
      "the video length of Normal_Videos115_x264.mp4 is: 299\n",
      "the video length of Normal_Videos116_x264.mp4 is: 4531\n",
      "the video length of Normal_Videos117_x264.mp4 is: 1176\n",
      "the video length of Normal_Videos118_x264.mp4 is: 948\n",
      "the video length of Normal_Videos119_x264.mp4 is: 3869\n",
      "the video length of Normal_Videos120_x264.mp4 is: 1481\n",
      "the video length of Normal_Videos121_x264.mp4 is: 1158\n",
      "the video length of Normal_Videos122_x264.mp4 is: 1893\n",
      "the video length of Normal_Videos124_x264.mp4 is: 1394\n",
      "the video length of Normal_Videos125_x264.mp4 is: 3121\n",
      "the video length of Normal_Videos126_x264.mp4 is: 2808\n",
      "the video length of Normal_Videos127_x264.mp4 is: 1789\n",
      "the video length of Normal_Videos128_x264.mp4 is: 1910\n",
      "the video length of Normal_Videos131_x264.mp4 is: 1894\n",
      "the video length of Normal_Videos132_x264.mp4 is: 2627\n",
      "the video length of Normal_Videos133_x264.mp4 is: 4330\n",
      "the video length of Normal_Videos134_x264.mp4 is: 4476\n",
      "the video length of Normal_Videos135_x264.mp4 is: 1787\n",
      "the video length of Normal_Videos140_x264.mp4 is: 1795\n",
      "the video length of Normal_Videos142_x264.mp4 is: 1871\n",
      "the video length of Normal_Videos143_x264.mp4 is: 1789\n",
      "the video length of Normal_Videos145_x264.mp4 is: 4504\n",
      "the video length of Normal_Videos146_x264.mp4 is: 817\n",
      "the video length of Normal_Videos147_x264.mp4 is: 509\n",
      "the video length of Normal_Videos148_x264.mp4 is: 625\n",
      "the video length of Normal_Videos149_x264.mp4 is: 1021\n",
      "the video length of Normal_Videos154_x264.mp4 is: 742\n",
      "the video length of Normal_Videos155_x264.mp4 is: 210\n",
      "the video length of Normal_Videos156_x264.mp4 is: 744\n",
      "the video length of Normal_Videos157_x264.mp4 is: 203\n",
      "the video length of Normal_Videos158_x264.mp4 is: 541\n",
      "the video length of Normal_Videos159_x264.mp4 is: 1080\n",
      "the video length of Normal_Videos160_x264.mp4 is: 3600\n",
      "the video length of Normal_Videos162_x264.mp4 is: 1169\n",
      "the video length of Normal_Videos163_x264.mp4 is: 3304\n",
      "the video length of Normal_Videos165_x264.mp4 is: 288\n",
      "the video length of Normal_Videos166_x264.mp4 is: 1607\n",
      "the video length of Normal_Videos167_x264.mp4 is: 1579\n",
      "the video length of Normal_Videos169_x264.mp4 is: 1554\n",
      "the video length of Normal_Videos170_x264.mp4 is: 1086\n",
      "the video length of Normal_Videos173_x264.mp4 is: 600\n",
      "the video length of Normal_Videos174_x264.mp4 is: 926\n",
      "the video length of Normal_Videos178_x264.mp4 is: 1028\n",
      "the video length of Normal_Videos181_x264.mp4 is: 4136\n",
      "the video length of Normal_Videos183_x264.mp4 is: 1680\n",
      "the video length of Normal_Videos184_x264.mp4 is: 1927\n",
      "the video length of Normal_Videos190_x264.mp4 is: 3627\n",
      "the video length of Normal_Videos191_x264.mp4 is: 3595\n",
      "the video length of Normal_Videos192_x264.mp4 is: 4021\n",
      "the video length of Normal_Videos193_x264.mp4 is: 3565\n",
      "the video length of Normal_Videos194_x264.mp4 is: 4141\n",
      "the video length of Normal_Videos195_x264.mp4 is: 4755\n",
      "the video length of Normal_Videos197_x264.mp4 is: 2465\n",
      "the video length of Normal_Videos198_x264.mp4 is: 2242\n",
      "the video length of Normal_Videos200_x264.mp4 is: 3706\n",
      "the video length of Normal_Videos201_x264.mp4 is: 2849\n",
      "the video length of Normal_Videos202_x264.mp4 is: 2495\n",
      "the video length of Normal_Videos204_x264.mp4 is: 824\n",
      "the video length of Normal_Videos207_x264.mp4 is: 2609\n",
      "the video length of Normal_Videos208_x264.mp4 is: 2250\n",
      "the video length of Normal_Videos209_x264.mp4 is: 633\n",
      "the video length of Normal_Videos213_x264.mp4 is: 1786\n",
      "the video length of Normal_Videos215_x264.mp4 is: 883\n",
      "the video length of Normal_Videos216_x264.mp4 is: 3420\n",
      "the video length of Normal_Videos218_x264.mp4 is: 1233\n",
      "the video length of Normal_Videos219_x264.mp4 is: 2670\n",
      "the video length of Normal_Videos220_x264.mp4 is: 2733\n",
      "the video length of Normal_Videos221_x264.mp4 is: 2229\n",
      "the video length of Normal_Videos222_x264.mp4 is: 949\n",
      "the video length of Normal_Videos223_x264.mp4 is: 4232\n",
      "the video length of Normal_Videos227_x264.mp4 is: 1657\n",
      "the video length of Normal_Videos228_x264.mp4 is: 1803\n",
      "the video length of Normal_Videos229_x264.mp4 is: 1827\n"
     ]
    },
    {
     "ename": "KeyboardInterrupt",
     "evalue": "",
     "output_type": "error",
     "traceback": [
      "\u001b[1;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[1;31mKeyboardInterrupt\u001b[0m                         Traceback (most recent call last)",
      "\u001b[1;32m<ipython-input-8-90cec4dcb0d5>\u001b[0m in \u001b[0;36m<module>\u001b[1;34m\u001b[0m\n\u001b[0;32m     35\u001b[0m     \u001b[0mvideo_path\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;34m\"E:/UCF-Anomaly-Detection-Dataset/UCF_Crimes/Videos/Training_Normal_Videos_Anomaly/\"\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0;32m     36\u001b[0m     \u001b[0mframe_path\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;34m\"G:/anomaly/project/data/small_video/normal/\"\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[1;32m---> 37\u001b[1;33m     \u001b[0mvideo2frame_normal\u001b[0m\u001b[1;33m(\u001b[0m\u001b[0mvideo_path\u001b[0m\u001b[1;33m,\u001b[0m\u001b[0mframe_path\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;32mNone\u001b[0m\u001b[1;33m,\u001b[0m\u001b[1;36m5\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0m",
      "\u001b[1;32m<ipython-input-8-90cec4dcb0d5>\u001b[0m in \u001b[0;36mvideo2frame_normal\u001b[1;34m(video_path, frame_path, numOfFrame, sample_rate)\u001b[0m\n\u001b[0;32m     21\u001b[0m         \u001b[0mk\u001b[0m \u001b[1;33m=\u001b[0m \u001b[1;36m0\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0;32m     22\u001b[0m         \u001b[1;32mwhile\u001b[0m \u001b[1;32mTrue\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[1;32m---> 23\u001b[1;33m             \u001b[0mret\u001b[0m\u001b[1;33m,\u001b[0m\u001b[0mimage\u001b[0m \u001b[1;33m=\u001b[0m \u001b[0mvideo\u001b[0m\u001b[1;33m.\u001b[0m\u001b[0mread\u001b[0m\u001b[1;33m(\u001b[0m\u001b[1;33m)\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0m\u001b[0;32m     24\u001b[0m             \u001b[1;32mif\u001b[0m \u001b[0mimage\u001b[0m \u001b[1;32mis\u001b[0m \u001b[1;32mNone\u001b[0m\u001b[1;33m:\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n\u001b[0;32m     25\u001b[0m                 \u001b[1;32mbreak\u001b[0m\u001b[1;33m\u001b[0m\u001b[1;33m\u001b[0m\u001b[0m\n",
      "\u001b[1;31mKeyboardInterrupt\u001b[0m: "
     ]
    }
   ],
   "source": [
    "def video2frame_normal(video_path,frame_path,numOfFrame,sample_rate):\n",
    "    file_list = os.listdir(video_path)\n",
    "    i = 0\n",
    "    for filename in file_list:\n",
    "        video = cv.VideoCapture(video_path + str(filename))\n",
    "        os.mkdir(os.path.join(frame_path,os.path.join(\"positive\"+str(i))))\n",
    "        video_length = video.get(7)\n",
    "        if video_length > 5000:\n",
    "            i += 1\n",
    "            continue\n",
    "        if sample_rate == None:\n",
    "            sample_rate = (video_length // numOfFrame)\n",
    "            if sample_rate == 0:\n",
    "                sample_rate = 1\n",
    "        if numOfFrame == None:\n",
    "            if video_length <= 700:\n",
    "                sample_rate = int(sample_rate // 2)\n",
    "            numOfFrame = int((video_length // sample_rate))\n",
    "        print(\"the video length of %s is: %d\"%(str(filename),video_length))\n",
    "        \n",
    "        k = 0\n",
    "        while True:\n",
    "            ret,image = video.read()\n",
    "            if image is None:\n",
    "                break\n",
    "            if k % sample_rate == 0:\n",
    "                image_name = frame_path + \"positive\" + str(i) + '/' + str(int(k//sample_rate)) + '.jpg'\n",
    "                cv.imwrite(image_name,image)\n",
    "            k += 1\n",
    "            if (k / numOfFrame) == sample_rate or k == 30 * sample_rate:\n",
    "                break\n",
    "    \n",
    "        i += 1\n",
    "if __name__ == \"__main__\":\n",
    "    video_path = \"E:/UCF-Anomaly-Detection-Dataset/UCF_Crimes/Videos/Training_Normal_Videos_Anomaly/\"\n",
    "    frame_path = \"G:/anomaly/project/data/small_video/normal/\"\n",
    "    video2frame_normal(video_path,frame_path,None,5)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [],
   "source": [
    "#this fn will rename all the videos in the folder, the new video name will be: new_name+<video number>\n",
    "def video_rename(path,new_name):\n",
    "    count = 200\n",
    "    file_list = os.listdir(path)\n",
    "    for file in file_list:\n",
    "        olddir = os.path.join(path,file)\n",
    "        filename = os.path.splitext(file)[0]              \n",
    "        filetype = os.path.splitext(file)[1]\n",
    "        newdir = os.path.join(path+new_name+str(count)+filetype)\n",
    "        os.rename(olddir,newdir)\n",
    "        count += 1\n",
    "\n",
    "path =  \"G:/anomaly/project/data/small_video_1/normal/\"\n",
    "new_name = \"negative\"\n",
    "video_rename(path,new_name)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Moviepy - Building video C:/Users/18819/Videos/QQplayerClip/111.mp4.\n",
      "MoviePy - Writing audio in 111TEMP_MPY_wvf_snd.mp3\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "                                                                                                                       \r"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "MoviePy - Done.\n",
      "Moviepy - Writing video C:/Users/18819/Videos/QQplayerClip/111.mp4\n",
      "\n"
     ]
    },
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "                                                                                                                       \r"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Moviepy - Done !\n",
      "Moviepy - video ready C:/Users/18819/Videos/QQplayerClip/111.mp4\n"
     ]
    }
   ],
   "source": [
    "# this fn changes the fps of the video\n",
    "def video_changeFPS(video_path,output_path):\n",
    "    clip = VideoFileClip(video_path)\n",
    "    clip.write_videofile(output_path, fps=120)\n",
    "    clip.reader.close()\n",
    "\n",
    "video_path = \"C:/Users/18819/Videos/QQplayerClip/1.mp4\"\n",
    "output_path = \"C:/Users/18819/Videos/QQplayerClip/111.mp4\"\n",
    "video_changeFPS(video_path,output_path)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "prosessing:  positive0\n",
      "prosessing:  positive1\n",
      "prosessing:  positive10\n",
      "prosessing:  positive103\n",
      "prosessing:  positive106\n",
      "prosessing:  positive107\n",
      "prosessing:  positive108\n",
      "prosessing:  positive109\n",
      "prosessing:  positive11\n",
      "prosessing:  positive110\n",
      "prosessing:  positive115\n",
      "prosessing:  positive117\n",
      "prosessing:  positive118\n",
      "prosessing:  positive12\n",
      "prosessing:  positive120\n",
      "prosessing:  positive121\n",
      "prosessing:  positive122\n",
      "prosessing:  positive129\n",
      "prosessing:  positive13\n",
      "prosessing:  positive130\n",
      "prosessing:  positive131\n",
      "prosessing:  positive132\n",
      "prosessing:  positive133\n",
      "prosessing:  positive134\n",
      "prosessing:  positive136\n",
      "prosessing:  positive139\n",
      "prosessing:  positive14\n",
      "prosessing:  positive140\n",
      "prosessing:  positive142\n",
      "prosessing:  positive146\n",
      "prosessing:  positive147\n",
      "prosessing:  positive15\n",
      "prosessing:  positive150\n",
      "prosessing:  positive153\n",
      "prosessing:  positive154\n",
      "prosessing:  positive155\n",
      "prosessing:  positive16\n",
      "prosessing:  positive160\n",
      "prosessing:  positive161\n",
      "prosessing:  positive162\n",
      "prosessing:  positive163\n",
      "prosessing:  positive165\n",
      "prosessing:  positive166\n",
      "prosessing:  positive167\n",
      "prosessing:  positive169\n",
      "prosessing:  positive170\n",
      "prosessing:  positive171\n",
      "prosessing:  positive172\n",
      "prosessing:  positive175\n",
      "prosessing:  positive177\n",
      "prosessing:  positive180\n",
      "prosessing:  positive183\n",
      "prosessing:  positive184\n",
      "prosessing:  positive186\n",
      "prosessing:  positive187\n",
      "prosessing:  positive189\n",
      "prosessing:  positive19\n",
      "prosessing:  positive192\n",
      "prosessing:  positive193\n",
      "prosessing:  positive2\n",
      "prosessing:  positive21\n",
      "prosessing:  positive23\n",
      "prosessing:  positive25\n",
      "prosessing:  positive26\n",
      "prosessing:  positive27\n",
      "prosessing:  positive28\n",
      "prosessing:  positive3\n",
      "prosessing:  positive30\n",
      "prosessing:  positive34\n",
      "prosessing:  positive35\n",
      "prosessing:  positive36\n",
      "prosessing:  positive37\n",
      "prosessing:  positive38\n",
      "prosessing:  positive39\n",
      "prosessing:  positive4\n",
      "prosessing:  positive42\n",
      "prosessing:  positive45\n",
      "prosessing:  positive47\n",
      "prosessing:  positive48\n",
      "prosessing:  positive5\n",
      "prosessing:  positive58\n",
      "prosessing:  positive59\n",
      "prosessing:  positive60\n",
      "prosessing:  positive63\n",
      "prosessing:  positive65\n",
      "prosessing:  positive69\n",
      "prosessing:  positive7\n",
      "prosessing:  positive70\n",
      "prosessing:  positive72\n",
      "prosessing:  positive73\n",
      "prosessing:  positive74\n",
      "prosessing:  positive75\n",
      "prosessing:  positive76\n",
      "prosessing:  positive8\n",
      "prosessing:  positive81\n",
      "prosessing:  positive83\n",
      "prosessing:  positive84\n",
      "prosessing:  positive85\n",
      "prosessing:  positive86\n",
      "prosessing:  positive88\n",
      "prosessing:  positive89\n",
      "prosessing:  positive9\n",
      "prosessing:  positive90\n",
      "prosessing:  positive91\n",
      "prosessing:  positive92\n",
      "prosessing:  positive93\n",
      "prosessing:  positive94\n",
      "prosessing:  positive95\n",
      "prosessing:  positive96\n",
      "prosessing:  positive97\n"
     ]
    }
   ],
   "source": [
    "# frame to video\n",
    "def frames2video(fps, save_path, frames_path):\n",
    "    fourcc = cv.VideoWriter_fourcc(*'mp4v')\n",
    "    videoWriter = cv.VideoWriter(save_path, fourcc, fps, (320, 240))\n",
    "    for i in range(1000):\n",
    "        if os.path.isfile(\"%s/%d.jpg\"%(frames_path, i)):\n",
    "            frame = cv.imread(\"%s/%d.jpg\"%(frames_path, i))\n",
    "            videoWriter.write(frame)\n",
    "    videoWriter.release()\n",
    "    return\n",
    " \n",
    "frame_path = \"G:/anomaly/project/data/small_video/normal/\"\n",
    "save_path =\"G:/anomaly/project/data/small_video_1/normal/\"\n",
    "file_list = os.listdir(frame_path)\n",
    "i = 0\n",
    "for fn in file_list:\n",
    "    print(\"prosessing: \",fn)\n",
    "    frames2video(1,os.path.join(save_path+\"negative\"+str(i)+\".mp4\"),\n",
    "                   os.path.join(frame_path+fn))\n",
    "    i += 1"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "(240, 320)\n"
     ]
    }
   ],
   "source": [
    "# feature test\n",
    "patt = \"G:/anomaly/project/data/image/more_frame_new/positive/positive33/imageP132.jpg\"\n",
    "image = cv.imread(patt)\n",
    "gray = cv.cvtColor(image, cv.COLOR_BGR2GRAY)\n",
    "canny = cv.Canny(gray, 0,200)\n",
    "print(canny.shape)\n",
    "canny_abs = np.uint8(np.absolute(canny))\n",
    "cv.imshow(\"Edge detection by Canny\", np.hstack([gray,canny_abs]))\n",
    "\n",
    "if(cv.waitKey(0)==27):\n",
    "    cv.destroyAllWindows()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [],
   "source": [
    "# test if your folder has 30 frames\n",
    "path = \"G:/anomaly/project/data/small_video/positive/\"\n",
    "file_list = os.listdir(path)\n",
    "for fn in file_list:\n",
    "    video_list = os.listdir(os.path.join(path + fn + \"/\"))\n",
    "    i = 0\n",
    "    for frame in video_list:\n",
    "        i += 1\n",
    "    if i != 30:\n",
    "        print(fn)    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [],
   "source": [
    "# some test code\n",
    "save_path =\"G:/anomaly/project/data/small_video_1/Abuse_Assault/positive/positive2.mp4\"\n",
    "video = cv.VideoCapture(save_path)\n",
    "l = video.get(7)\n",
    "k = 0\n",
    "while True:\n",
    "    ret,image = video.read()\n",
    "    if image is None:\n",
    "        break\n",
    "    image_name = \"G:/test/\" + str(k) + \".jpg\"\n",
    "    cv.imwrite(image_name,image)\n",
    "    k += 1"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "0.0"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "l"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "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.8.3"
  },
  "toc": {
   "base_numbering": 1,
   "nav_menu": {},
   "number_sections": true,
   "sideBar": true,
   "skip_h1_title": false,
   "title_cell": "Table of Contents",
   "title_sidebar": "Contents",
   "toc_cell": false,
   "toc_position": {},
   "toc_section_display": true,
   "toc_window_display": false
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}