Broiler_Detection.ipynb

{"cells":[{"cell_type":"code","execution_count":null,"metadata":{"colab":{"background_save":true,"base_uri":"https://localhost:8080/","output_embedded_package_id":"1Vktrx97koPjkkAF0bhMNh9E933sX-_3K"},"id":"e79qAhhkBMT2","outputId":"83cba4b4-0374-487c-8aa6-28da180316f1"},"outputs":[],"source":["######### Libraries and Directories\n","import tensorflow as tf\n","import numpy as np\n","import pandas as pd\n","import matplotlib.pyplot as plt\n","from sklearn.utils import shuffle\n","from keras.utils import np_utils\n","import cv2 as cv\n","import cv2\n","from random import randint\n","from google.colab import drive\n","drive.mount('/content/drive')\n","import os\n","os.getcwd()\n","os.chdir('/content/drive/MyDrive/Colab Notebooks/ICCKE2022_Challenge/yolo-v4-tf.keras')\n","from utils import DataGenerator, read_annotation_lines\n","from models import Yolov4\n","from config import yolo_config\n","######### Libraries and Directories\n","\n","######### Converting the xml Annotation file to txt Annotation file\n","#!python xml_to_txt.py\n","######### Converting the xml Annotation file to txt Annotation file\n","\n","######### Setting Yolov4 for Transferring Learning and Fine Tuning\n","#### Directory of Annotations\n","train_lines, val_lines = read_annotation_lines('/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/Annotation/anno3.txt', test_size=0.1)\n","#### Directory of Trainig Images\n","FOLDER_PATH = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img'\n","#### Directory of Trainig Classes\n","class_name_path = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/Classes/classes.txt'\n","#### Generating the Training Data for Yolov4\n","data_gen_train = DataGenerator(train_lines, class_name_path, FOLDER_PATH)\n","data_gen_val = DataGenerator(val_lines, class_name_path, FOLDER_PATH)\n","#### Directory of Yolov4 Weights for Fine Tuning\n","weight_path = '/content/drive/My Drive/Colab Notebooks/darknet/yolov4.conv.137'\n","######### Setting Yolov4 for Transferring Learning and Fine Tuning\n","\n","###### Creating a new model and Start Training\n","model = Yolov4(weight_path=weight_path,class_name_path=class_name_path)\n","h = model.fit(data_gen_train,\n","          initial_epoch=0,\n","          epochs=1000,\n","          val_data_gen=data_gen_val,\n","          callbacks=[])\n","#model.save_model('/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/final_training')\n","###### Creating a new model and Start Training\n","'''\n","###### Loading the Stored Model\n","path = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/final_training'\n","my_model = Yolov4(weight_path=weight_path,\n","               class_name_path=class_name_path)\n","my_model.load_model(path)\n","###### Loading the Stored Model\n","'''\n","###### Testing the Trained Model For Broiler Detection\n","img_path1 = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img_test/frame00097.png'\n","img_path2 = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img_test/frame00201.png'\n","img_path3 = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img_test/frame00289.png'\n","img_path4 = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img_test/frame00301.png'\n","img_path5 = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img_test/frame00601.png'\n","img_path6 = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img_test/frame00801.png'\n","img_path7 = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img_test/frame00901.png'\n","img_path8 = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img_test/frame01201.png'\n","img_path9 = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img_test/frame01001.png'\n","img_path10 = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img_test/frame01301.png'\n","img_path11 = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img_test/frame01501.png'\n","img_path12 = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img_test/frame01701.png'\n","img_path13 = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img_test/frame02601.png'\n","img_path14 = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img_test/frame02901.png'\n","img_path15 = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img_test/frame03901.png'\n","img_path16 = '/content/drive/My Drive/Colab Notebooks/ICCKE2022_Challenge/img_test/frame04101.png'\n","my_model.predict(img_path1)\n","my_model.predict(img_path2)\n","my_model.predict(img_path3)\n","my_model.predict(img_path4)\n","my_model.predict(img_path5)\n","my_model.predict(img_path6)\n","my_model.predict(img_path7)\n","my_model.predict(img_path8)\n","my_model.predict(img_path9)\n","my_model.predict(img_path10)\n","my_model.predict(img_path11)\n","my_model.predict(img_path12)\n","my_model.predict(img_path13)\n","my_model.predict(img_path14)\n","my_model.predict(img_path15)\n","my_model.predict(img_path16)\n","###### Testing the Trained Model For Broiler Detection\n","\n"]}],"metadata":{"accelerator":"GPU","colab":{"machine_shape":"hm","name":"","provenance":[{"file_id":"1FHxYx4XYVgf8lXYzvZqolievghtZdUJX","timestamp":1666824849190},{"file_id":"1ml7SYy-OfldreOKdW1wLQwNMyEbOFx6O","timestamp":1665786498949},{"file_id":"1dHUe-pL5RoIFM7l4ShWek_jsr09I6_Gn","timestamp":1618083089547},{"file_id":"1MXKLy9_0IaAVmMYbaEyshvnsihuQt5c5","timestamp":1610064699028},{"file_id":"1ljVaGMDVuQqKb2DwdTGzvW_M8buwQPRC","timestamp":1609513687429},{"file_id":"1TKYnUYcQNsuaMtr-j04IPvenUQwmrjVd","timestamp":1609337647023},{"file_id":"155S_bb3viIoL0wAwkIyr1r8XQu4ARwA9","timestamp":1608291787201}],"version":""},"gpuClass":"standard","kernelspec":{"display_name":"Python 3","name":"python3"}},"nbformat":4,"nbformat_minor":0}