YOLOv3_mask_detect.ipynb

{"nbformat":4,"nbformat_minor":0,"metadata":{"colab":{"name":"YOLOv3_mask_detect","provenance":[{"file_id":"1-93cehpSEAaujWFDMZxTgrAaXxDIxZTd","timestamp":1583652317261}],"private_outputs":true,"collapsed_sections":[],"authorship_tag":"ABX9TyMeJlSxQ+r1GxFonF/VakAk"},"kernelspec":{"name":"python3","display_name":"Python 3"},"accelerator":"GPU"},"cells":[{"cell_type":"markdown","metadata":{"id":"q-NeipssA6lj","colab_type":"text"},"source":["## Connect the Colab notebook to Google Drive\n","\n"]},{"cell_type":"code","metadata":{"id":"72c9PYPk9Iuh","colab_type":"code","colab":{}},"source":["# Connect the Colab notebook to Google Drive\n","from google.colab import drive\n","drive.mount('/content/gdrive')"],"execution_count":0,"outputs":[]},{"cell_type":"code","metadata":{"id":"upDNPzS4_BgX","colab_type":"code","colab":{}},"source":["!sudo apt-get install tree"],"execution_count":0,"outputs":[]},{"cell_type":"code","metadata":{"id":"ftQDPNa9AKIF","colab_type":"code","colab":{}},"source":["# tree command\n","!tree /content/gdrive/My\\ Drive/Python/darknet"],"execution_count":0,"outputs":[]},{"cell_type":"markdown","metadata":{"id":"xtcCUQQ3BlOh","colab_type":"text"},"source":["## Check CUDA release version"]},{"cell_type":"code","metadata":{"id":"aqAc6CIrAXvV","colab_type":"code","colab":{}},"source":["!/usr/local/cuda/bin/nvcc --version"],"execution_count":0,"outputs":[]},{"cell_type":"markdown","metadata":{"id":"zx6Fmf26EM9E","colab_type":"text"},"source":["## Install cuDNN according to the current CUDA version"]},{"cell_type":"code","metadata":{"id":"RaPoNk-SEOfh","colab_type":"code","colab":{}},"source":["# CUDA 10.0  cuDNN為 v7.6.5\n","# CUDA 10.1  cuDNN為 v7.6.5\n","\n","# cuDNN https://developer.nvidia.com/rdp/cudnn-download\n","\n","# On your local computer, create a folder named cuDNN in your local folder darknet. Copy the tgz file there"],"execution_count":0,"outputs":[]},{"cell_type":"code","metadata":{"id":"VnRoMd0PEdPl","colab_type":"code","colab":{}},"source":["!tar -xzvf gdrive/My\\ Drive/Python/darknet/cuDNN/cudnn-10.0-linux-x64-v7.6.5.32.tgz -C /usr/local/\n","# !tar -xzvf gdrive/My\\ Drive/Python/darknet/cuDNN/cudnn-10.1-linux-x64-v7.6.5.32.tgz -C /usr/local/\n","\n","!chmod a+r /usr/local/cuda/include/cudnn.h\n","\n","!cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2"],"execution_count":0,"outputs":[]},{"cell_type":"markdown","metadata":{"id":"v2LHUXi_J0qQ","colab_type":"text"},"source":["# Cloning and compiling Darkent."]},{"cell_type":"code","metadata":{"id":"D0YLHV-qJ3u-","colab_type":"code","colab":{}},"source":["!git clone https://github.com/kriyeng/darknet/\n","%cd darknet\n","\n","!git checkout feature/google-colab\n","\n","#Compile Darknet\n","!make\n","\n","#Copies the Darknet compiled version to Google drive\n","# !cp ./darknet /content/gdrive/My\\ Drive/Python/darknet/bin/darknet"],"execution_count":0,"outputs":[]},{"cell_type":"markdown","metadata":{"id":"ubE0cHSPNR_F","colab_type":"text"},"source":["## Copying the compiled version of Darknet from Drive."]},{"cell_type":"code","metadata":{"id":"mkZiOmnMNTFQ","colab_type":"code","colab":{}},"source":["# Uncomment after the first run, when you have a copy of compiled darkent in your Google Drive\n","\n","# Makes a dir for darknet and move there\n","# !mkdir darknet\n","# %cd darknet\n","\n","# Copy the Darkent compiled version to the VM local drive\n","# !cp /content/gdrive/My\\ Drive/Python/darknet/bin/darknet ./darknet\n","\n","# !chmod +x ./darknet"],"execution_count":0,"outputs":[]},{"cell_type":"markdown","metadata":{"id":"5Eo70UyVYrbb","colab_type":"text"},"source":["## Some Utils\n","#### imShow/upload/download"]},{"cell_type":"code","metadata":{"id":"35Nuv9qmZpZY","colab_type":"code","colab":{}},"source":["# download files \n","def imShow(path):\n","  import cv2\n","  import matplotlib.pyplot as plt\n","  %matplotlib inline\n","\n","  image = cv2.imread(path)\n","  height, width = image.shape[:2]\n","  resized_image = cv2.resize(image,(3*width, 3*height), interpolation = cv2.INTER_CUBIC)\n","\n","  fig = plt.gcf()\n","  fig.set_size_inches(18, 10)\n","  plt.axis(\"off\")\n","  #plt.rcParams['figure.figsize'] = [10, 5]\n","  plt.imshow(cv2.cvtColor(resized_image, cv2.COLOR_BGR2RGB))\n","  plt.show()\n","\n","\n","def upload():\n","  from google.colab import files\n","  uploaded = files.upload() \n","  for name, data in uploaded.items():\n","    with open(name, 'wb') as f:\n","      f.write(data)\n","      print ('saved file', name)\n","\n","\n","def download(path):\n","  from google.colab import files\n","  files.download(path)"],"execution_count":0,"outputs":[]},{"cell_type":"code","metadata":{"id":"J3YEQ0vQDT3C","colab_type":"code","colab":{}},"source":["# copy draknet directory to google colab\n","!cp -r /content/gdrive/My\\ Drive/Python/darknet/ /content/darknet/"],"execution_count":0,"outputs":[]},{"cell_type":"markdown","metadata":{"id":"KfaWIQ4XD2lJ","colab_type":"text"},"source":["## Test darknet"]},{"cell_type":"code","metadata":{"id":"BEeEinJLFrUl","colab_type":"code","colab":{}},"source":["# Get yolov3 weights\n","!wget https://pjreddie.com/media/files/yolov3.weights"],"execution_count":0,"outputs":[]},{"cell_type":"code","metadata":{"id":"qM-zlXL5b4RN","colab_type":"code","colab":{}},"source":["# Test\n","!./darknet detect cfg/yolov3.cfg weights/yolov3.weights data/person.jpg -dont-show\n","imShow('predictions.jpg')"],"execution_count":0,"outputs":[]},{"cell_type":"markdown","metadata":{"id":"zXDpR-VBse3E","colab_type":"text"},"source":["## Mask"]},{"cell_type":"code","metadata":{"id":"vFhd19aWskML","colab_type":"code","colab":{}},"source":["# Mask training\n","!./darknet detector train \"/content/darknet/Mask/obj.data\" \"/content/darknet/Mask/yolov3_mask.cfg\" \"/content/darknet/weights/darknet53.conv.74\" -dont_show"],"execution_count":0,"outputs":[]},{"cell_type":"code","metadata":{"id":"kEdKDBfRaq7_","colab_type":"code","colab":{}},"source":["# photo detect\n","# !./darknet detect Mask/yolov3_mask.cfg backup/yolov3_mask_last.weights Mask/yolo/Mask_121.jpg -thresh 0\n","!./darknet detect Mask/yolov3_mask.cfg Mask//yolov3_mask_last.weights Mask/yolo/Mask_121.jpg\n","imShow('predictions.jpg')"],"execution_count":0,"outputs":[]},{"cell_type":"code","metadata":{"id":"O6LrFe_np3oJ","colab_type":"code","colab":{}},"source":["# video detect\n","!./darknet detector demo /content/darknet/Mask/obj.data Mask/yolov3_mask.cfg /content/darknet/weights/yolov3_mask_last.weights -dont_show /content/darknet/Mask/Mask_test_v_3.mp4 -i 0 -out_filename predictions.avi"],"execution_count":0,"outputs":[]}]}