MainJupyter.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Linear Barcodes Verification Project - Computer Vision and Image Processing M\n",
    "\n",
    "#### Realizzato da:\n",
    "    - Arianna Solini\n",
    "    - Massimo Schembri\n",
    "    - Riccardo Vasumini"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "scrolled": true
   },
   "outputs": [],
   "source": [
    "import cv2\n",
    "import glob\n",
    "import matplotlib.pyplot as plt\n",
    "from colorama import Fore, Style\n",
    "from barcode import Barcode\n",
    "\n",
    "title = r\"\"\"██████╗  █████╗ ██████╗  ██████╗ ██████╗ ██████╗ ███████╗     █████╗ ███╗   ██╗ █████╗ ██╗  ██╗   ██╗███████╗███████╗██████╗ \n",
    "██╔══██╗██╔══██╗██╔══██╗██╔════╝██╔═══██╗██╔══██╗██╔════╝    ██╔══██╗████╗  ██║██╔══██╗██║  ╚██╗ ██╔╝╚══███╔╝██╔════╝██╔══██╗\n",
    "██████╔╝███████║██████╔╝██║     ██║   ██║██║  ██║█████╗      ███████║██╔██╗ ██║███████║██║   ╚████╔╝   ███╔╝ █████╗  ██████╔╝\n",
    "██╔══██╗██╔══██║██╔══██╗██║     ██║   ██║██║  ██║██╔══╝      ██╔══██║██║╚██╗██║██╔══██║██║    ╚██╔╝   ███╔╝  ██╔══╝  ██╔══██╗\n",
    "██████╔╝██║  ██║██║  ██║╚██████╗╚██████╔╝██████╔╝███████╗    ██║  ██║██║ ╚████║██║  ██║███████╗██║   ███████╗███████╗██║  ██║\n",
    "╚═════╝ ╚═╝  ╚═╝╚═╝  ╚═╝ ╚═════╝ ╚═════╝ ╚═════╝ ╚══════╝    ╚═╝  ╚═╝╚═╝  ╚═══╝╚═╝  ╚═╝╚══════╝╚═╝   ╚══════╝╚══════╝╚═╝  ╚═╝\n",
    "                                                                                                                             \"\"\"\n",
    "\n",
    "def print_title():\n",
    "    print(Fore.YELLOW + title)\n",
    "    print(Style.RESET_ALL)\n",
    "    \n",
    "def main(barcode_directory):\n",
    "\n",
    "    column_names = [\"IMAGE NAME\", \"BARCODE GRADE\", \"X-DIMENSION\", \"HEIGHT\", \"BOUNDING BOX\", \"ROTATION ANGLE\",\n",
    "                        \"SCANLINE-1\",\"SCANLINE-2\", \"SCANLINE-3\", \"SCANLINE-4\", \"SCANLINE-5\", \n",
    "                        \"SCANLINE-6\", \"SCANLINE-7\", \"SCANLINE-8\", \"SCANLINE-9\", \"SCANLINE-10\", \"SIZE BARS-SPACES\\n\"]\n",
    "    header = ','.join(column_names)\n",
    "    f = open('barcode_features.csv', 'w')\n",
    "    f.write(header)\n",
    "\n",
    "    #Cycle over the given images in the ./resources folder\n",
    "    for file in glob.glob(barcode_directory + \"/*.*\"):\n",
    "        try:\n",
    "\n",
    "            #print()\n",
    "            #print('|' * 80)\n",
    "            #print()\n",
    "\n",
    "            #Print the image_name and the image itself  \n",
    "            barcode = Barcode()\n",
    "            barcode.image_name = file\n",
    "            #print('image_name: ', barcode.image_name)\n",
    "\n",
    "            plt.title('ORIGINAL IMAGE', pad=10)\n",
    "            barcode.image = cv2.imread(barcode.image_name, cv2.IMREAD_GRAYSCALE)\n",
    "            plt.imshow(barcode.image, cmap = 'gray', vmin = 0, vmax = 255)\n",
    "            plt.show(block = True)\n",
    "\n",
    "            #Show the roi and print its coordinates\n",
    "            barcode.bin_image = cv2.adaptiveThreshold(barcode.image, 255, cv2.ADAPTIVE_THRESH_MEAN_C, cv2.THRESH_BINARY_INV, 41, 11)\n",
    "            barcode.set_roi()\n",
    "            plt.title('ORIGINAL ROI', pad=10)\n",
    "            plt.imshow(barcode.roi, cmap = 'gray', vmin = 0, vmax = 255)\n",
    "            plt.show(block = True)\n",
    "            #print('coord_roi: ', barcode.coord_roi)\n",
    "\n",
    "            #Show the roi after the verticalization\n",
    "            plt.title('VERTICALIZED ROI', pad=10)\n",
    "            barcode.verticalize_roi()\n",
    "            plt.imshow(barcode.roi, cmap = 'gray', vmin = 0, vmax = 255)\n",
    "            plt.show(block = True)\n",
    "\n",
    "            #Show the highlighted bars in red and print the minimum width of a bar and the minimum_height of a bar\n",
    "            barcode.bin_otsu_roi = cv2.threshold(barcode.roi, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)[1]\n",
    "            barcode.set_minimum_width_height_bars()\n",
    "            #print('min_width: %3.2f' % barcode.min_width)\n",
    "            #print('min_height: ', barcode.min_height)\n",
    "\n",
    "            #Show the roi resized with the following rule: bounding box enclosing the barcode plus a surrounding background area\n",
    "            #(quite zone) of size (quite zone): X dimension, above and below the code, 10*X before the first bar, 10* X following the last bar\n",
    "            plt.title('RESIZED ROI', pad=10)\n",
    "            barcode.resize_roi()\n",
    "            #Show the roi resized with the following rule\n",
    "            plt.imshow(barcode.roi, cmap = 'gray', vmin = 0, vmax = 255)\n",
    "            plt.show(block = True)\n",
    "\n",
    "            #Numbers and text elements are remmoved so to create a smaller box that would not include them.\n",
    "            plt.title('CUT ROI', pad=10)\n",
    "            barcode.cut_roi()\n",
    "            plt.imshow(barcode.roi, cmap = 'gray', vmin = 0, vmax = 255)\n",
    "            plt.show(block = True)\n",
    "\n",
    "            #Set the scanlines we have to analyse and plot them    \n",
    "            barcode.set_scanlines()\n",
    "\n",
    "            #Methods which sets the scanlines' parameters we have to find\n",
    "            barcode.set_scanlines_min_reflectance()\n",
    "\n",
    "            barcode.set_scanlines_max_reflectance()\n",
    "\n",
    "            barcode.set_scanlines_symbol_contrast()\n",
    "\n",
    "            barcode.set_scanlines_edges()\n",
    "\n",
    "            barcode.set_scalines_minimum_edge_contrast()\n",
    "\n",
    "            barcode.set_scanline_defects()\n",
    "\n",
    "            barcode.set_scanline_modulation()\n",
    "\n",
    "            barcode.set_left_to_right_widths()\n",
    "\n",
    "            #Set the grade of each scanline\n",
    "            barcode.grade_scanlines()\n",
    "\n",
    "            #Set the grade of the barcode\n",
    "            barcode.grade_barcode()\n",
    "\n",
    "            row = ','.join(barcode.write_row_lists())\n",
    "\n",
    "            f.write(row)\n",
    "        except:\n",
    "            print(\"The analysis didn't succeded\")\n",
    "            continue\n",
    "        \n",
    "    f.close()\n",
    "    \n",
    "if __name__ == \"__main__\":\n",
    "    print_title()\n",
    "    barcode_directory = input('Insert the name of the directory you want to analyse:\\t')\n",
    "    main(barcode_directory)\n",
    "    print(\"\\nEnd of analysis\")\n"
   ]
  },
  {
   "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.6.9"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}