Upload 1519 files

app/app.py

@@ -0,0 +1,81 @@
+import os
+import cv2
+import numpy as np
+ 
+
+from flask import Flask, render_template, request, jsonify, redirect, url_for
+from werkzeug.utils import secure_filename
+ 
+
+from pyimagesearch.colordescriptor import ColorDescriptor
+from pyimagesearch.searcher import Searcher
+
+
+# create flask instance
+app = Flask(__name__)
+
+INDEX = os.path.join(os.path.dirname(__file__), 'index.csv')
+
+# main route
+@app.route('/')
+def index():
+    return render_template('index.html', preview="static/init-preview.png")
+
+# image database url list route
+@app.route('/list', methods=['POST'])
+def image_list():
+
+    if request.method == "POST":
+
+        try:
+
+            imgList = [img for img in list(os.listdir(os.path.join(os.path.dirname(__file__), 'static/images/'))) if img[-4:] in ('.png', '.jpg', '.gif')]
+
+            return jsonify(imgList=imgList)
+        
+        except Exception as e:
+            return jsonify({"sorry": "Sorry, no results! Please try again."}), 500
+
+
+# search route
+@app.route('/search', methods=['POST'])
+def search():
+ 
+    if request.method == "POST":
+
+        RESULTS_ARRAY = []
+
+        # get url
+        image_url = request.form.get('img')
+ 
+        try:
+ 
+            # initialize the image descriptor
+            cd = ColorDescriptor((8, 12, 3))
+ 
+            # load the query image and describe it
+            from skimage import io
+            import cv2
+
+            query = cv2.imread(os.path.join(os.path.dirname(__file__), 'static/images/'+image_url))
+            features = cd.describe(query)
+ 
+            # perform the search
+            searcher = Searcher(INDEX)
+            results = searcher.search(features)
+ 
+            # loop over the results, displaying the score and image name
+            for (score, resultID) in results:
+                RESULTS_ARRAY.append(
+                    {"image": str(resultID), "score": str(score)})
+            # return success
+            return jsonify(results=(RESULTS_ARRAY[:101]), preview="images/"+image_url)
+ 
+        except Exception as e:
+            print(str(e))
+            # return error
+            return jsonify({"sorry": "Sorry, no results! Please try again."}), 500
+
+# run!
+if __name__ == '__main__':
+    app.run('0.0.0.0', debug=True)

app/index.py

@@ -0,0 +1,40 @@
+#import the necessary packages
+from pyimagesearch.colordescriptor import ColorDescriptor
+import argparse
+import glob
+import cv2
+
+#construct the argument parser and parse the arguments
+ap = argparse.ArgumentParser()
+ap.add_argument("-d", "--dataset", required=True,
+	help = "path to the directory that contains the images to be indexed")
+ap.add_argument("-i", "--index", required=True,
+	help = "path to where the index should be stored")
+args = vars(ap.parse_args())
+
+#initialize the color descriptor
+cd = ColorDescriptor((8, 12, 3))
+
+#open the output index file for writing
+output = open(args["index"], "w")
+types = ('/*.jpg', '/*.png', '/*.gif') # the tuple of file types
+files_grabbed = []
+for files in types:
+    files_grabbed.extend(glob.glob(args["dataset"]+files))
+
+#use glob to grab the image paths and loop over them
+for imagePath in files_grabbed:
+    #extract the imageID from the image
+    #path and load the image itself
+    imageID = imagePath[imagePath.rfind("/")+1:]
+    image = cv2.imread(imagePath)
+
+    #describe the image
+    features = cd.describe(image)
+
+    #write the features to file
+    features = [str(f) for f in features]
+    output.write("%s,%s\n" % (imageID, ",".join(features)))
+
+# close the index file
+output.close()

app/pyimagesearch/colordescriptor.py

@@ -0,0 +1,62 @@
+# import the necessary packages
+import numpy as np
+import cv2
+
+class ColorDescriptor:
+	def __init__(self, bins):
+		# store the number of bins for the 3D histogram
+		self.bins = bins
+
+	def describe(self, image):
+		# convert the image to the HSV color space and initialize
+		# the features used to quantify the image
+		image = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
+		features = []
+
+		# grab the dimensions and compute the center of the image
+		(h, w) = image.shape[:2]
+		(cX, cY) = (int(w * 0.5), int(h * 0.5))
+
+		# divide the image into four rectangles/segments (top-left,
+		# top-right, bottom-right, bottom-left)
+		segments = [(0, cX, 0, cY), (cX, w, 0, cY), (cX, w, cY, h),
+			(0, cX, cY, h)]
+ 
+		# construct an elliptical mask representing the center of the
+		# image
+		(axesX, axesY) = (int(w * 0.75) // 2, int(h * 0.75) // 2)
+		ellipMask = np.zeros(image.shape[:2], dtype = "uint8")
+		cv2.ellipse(ellipMask, (cX, cY), (axesX, axesY), 0, 0, 360, 255, -1)
+ 
+		# loop over the segments
+		for (startX, endX, startY, endY) in segments:
+			# construct a mask for each corner of the image, subtracting
+			# the elliptical center from it
+			cornerMask = np.zeros(image.shape[:2], dtype = "uint8")
+			cv2.rectangle(cornerMask, (startX, startY), (endX, endY), 255, -1)
+			cornerMask = cv2.subtract(cornerMask, ellipMask)
+ 
+			# extract a color histogram from the image, then update the
+			# feature vector
+			hist = self.histogram(image, cornerMask)
+			features.extend(hist)
+ 
+		# extract a color histogram from the elliptical region and
+		# update the feature vector
+		hist = self.histogram(image, ellipMask)
+		features.extend(hist)
+ 
+		# return the feature vector
+		return features
+
+	def histogram(self, image, mask):
+		# extract a 3D color histogram from the masked region of the
+		# image, using the supplied number of bins per channel; then
+		# normalize the histogram
+		hist = cv2.calcHist([image], [0, 1, 2], mask, self.bins,
+			[0, 180, 0, 256, 0, 256])
+		cv2.normalize(hist,hist)
+		hist = hist.flatten()
+ 
+		# return the histogram
+		return hist

app/pyimagesearch/searcher.py

@@ -0,0 +1,46 @@
+# import the necessary packages
+import numpy as np
+import csv
+
+class Searcher:
+	def __init__(self, indexPath):
+		#store our index path
+		self.indexPath = indexPath
+
+	def search(self, queryFeatures, limit=101):
+		#initialize dictionary of results
+		results = {}
+
+		#open the index file
+		with open(self.indexPath) as f:
+			#initialize the CSV reader
+			reader = csv.reader(f)
+
+			#loop over the rows in the index
+			for row in reader:
+				#parse out the image ID and features, then compute the
+				#chi-squared dist. b/w the features in our index
+				#and our query features
+				features = [float(x) for x in row[1:]]
+				d = self.chi2_distance(features, queryFeatures)
+
+				#Update dictionsary
+				#key is image id, value is similarity
+				results[row[0]] = d
+
+			#close reader
+			f.close()
+
+		#sort in order of relevance
+		results = sorted([(v,k) for (k,v) in results.items()])
+
+		#return our (limited) results
+		return results[:limit]
+
+	def chi2_distance(self, histA, histB, eps=1e-10):
+		#compute chi-squared distance
+		d = 0.5 * np.sum([((a-b)**2)/(a+b+eps)
+			for (a,b) in zip(histA, histB)])
+
+		# return the chi-squared distance
+		return d

app/requirements.txt

@@ -0,0 +1,25 @@
+appdirs==1.4.0
+certifi==2018.8.24
+click==6.7
+cycler==0.10.0
+dask==0.14.0
+decorator==4.0.11
+Flask==1.0.2
+gunicorn==19.7.0
+itsdangerous==0.24
+Jinja2==2.10.1
+MarkupSafe==1.0
+matplotlib==2.0.0
+networkx==1.11
+numpy==1.12.0
+olefile==0.44
+opencv-python==4.0.0.21
+packaging==16.8
+Pillow==6.2.1
+pyparsing==2.1.10
+python-dateutil==2.6.0
+pytz==2016.10
+scipy==0.19.0
+six==1.10.0
+toolz==0.8.2
+Werkzeug==0.15.3

app/static/main.css

@@ -0,0 +1,180 @@
+@import url('https://fonts.googleapis.com/css?family=Lato&display=swap');
+
+* {
+    box-sizing: border-box;
+    padding: 0;
+    margin: 0;
+    font-family: 'Lato', sans-serif;
+}
+
+body {
+    background-color: white;
+}
+
+.container {
+    width: 100%;
+}
+
+.header {
+    text-align: center;
+    font-weight: bold;
+    padding: 10px;
+    color: #0D19A3;
+    border-bottom: 5px solid #15DB95;
+}
+
+
+.display {
+    display: grid;
+    grid-template-columns: 1fr 3fr;
+    grid-column-gap: 20px;
+    margin-top: 10px;
+}
+
+.left-pane {
+    display: flex;
+    flex-direction: column;
+    align-items: center;
+    justify-content: flex-start;
+    color: #F4E4C1;
+    border-bottom: 3px solid #080F5B;
+    margin-left: 2px;
+}
+
+.img-preview {
+    width: 95%;
+    margin: 10px;
+    border: 4px solid #15DB95;
+}
+
+#preview-name {
+    color: #080F5B;
+    font-weight: bold;
+    font-size: 1.25em;
+    text-align: center;
+}
+
+#select {
+    padding: 10px;
+    background-color: #15DB95;
+    color: white;
+    font-weight: bold;
+    font-size: 1.5em;
+    cursor: pointer;
+    border-radius: 10px;
+    margin: 10px;
+    text-align: center;
+    transition: .3s;
+}
+
+#select:hover {
+    transform: scale(.95);
+    -webkit-box-shadow: 0px 0px 4px 3px rgba(8,15,91,1);
+    -moz-box-shadow: 0px 0px 4px 3px rgba(8,15,91,1);
+    box-shadow: 0px 0px 4px 3px rgba(8,15,91,1);
+}
+
+#results {
+    display: grid;
+    grid-template-columns: 1fr 1fr 1fr;
+    grid-column-gap: 15px;
+    grid-row-gap: 15px;
+    padding: 10px;
+}
+
+#searching {
+    font-weight: bold;
+    font-size: 1.75em;
+    color: #080F5B;
+}
+
+.img-result {
+    background-color: #080F5B;
+    color: white;
+    border: 2px solid #F4E4C1;
+    -webkit-box-shadow: 0px 0px 4px 3px rgba(8,15,91,1);
+    -moz-box-shadow: 0px 0px 4px 3px rgba(8,15,91,1);
+    box-shadow: 0px 0px 4px 3px rgba(8,15,91,1);
+}
+
+.img-info {
+    display: grid;
+    grid-template-columns: 1fr;
+    align-content: center;
+    justify-items: center;
+    border: 1px solid white;
+}
+
+.responsive {
+    width: 100%;
+    max-width: 100%;
+    height: 250px;
+}
+
+/* The Modal (background) */
+.modal-display {
+    display: none; /* Hidden by default */
+    position: fixed; /* Stay in place */
+    z-index: 1; /* Sit on top */
+    padding-top: 100px; /* Location of the box */
+    left: 0;
+    top: 0;
+    width: 100%; /* Full width */
+    height: 100%; /* Full height */
+    overflow: auto; /* Enable scroll if needed */
+    background-color: rgb(0,0,0); /* Fallback color */
+    background-color: rgba(0,0,0,0.4); /* Black w/ opacity */
+  }
+  
+  /* Modal Content */
+  .modal-content {
+    background-color: white;
+    margin: auto;
+    padding: 20px;
+    border: 2px solid black;
+    width: 90%;
+  }
+
+  .modal-images-list {
+    display: flex;
+    flex-wrap: wrap;
+    justify-content: space-between;
+  }
+
+  .modal-image {
+      margin-bottom: 10px;
+      width: 30%;
+      cursor: pointer;
+  }
+
+  .modal-image:hover {
+      border: 5px solid #0D19A3;
+  }
+  
+  /* The Close Button */
+  .close {
+    color: #aaaaaa;
+    float: right;
+    font-size: 28px;
+    font-weight: bold;
+    transform: translate(12px, -13px)
+  }
+  
+  .close:hover,
+  .close:focus {
+    color: #000;
+    text-decoration: none;
+    cursor: pointer;
+}
+
+@media (max-width: 500px){
+    .display {
+        grid-template-columns: 1fr;
+        grid-row-gap: 20px;
+    }
+
+    #results {
+        grid-template-columns: 1fr;
+        grid-row-gap: 20px;
+    }
+}

app/static/main.js

@@ -0,0 +1,113 @@
+// ----- custom js ----- //
+var imagePath = 'static/images/';
+
+$(document).ready(function() {
+
+  console.log("ready");
+  modalControl();
+  getAllImages();
+
+});
+
+// request for all images in database to display in modal
+var getAllImages = function(){
+
+     $.ajax({
+      url: "/list",
+      cache: false,
+      contentType: false,
+      processData: false,                        
+      type: 'POST',
+      success: function(data){
+        displayModalImages(data.imgList);
+      },
+      error: function(error){
+        console.log(error.toString()); 
+      }
+   });
+}
+
+// displays images in modal
+var displayModalImages = function(imgList){
+
+   for(var i = 0; i < imgList.length; i++){
+      $(".modal-images-list").append("<img src="+imagePath+imgList[i]+" class=modal-image onclick=imageSelectSearch(this) />");
+
+   }
+
+}
+
+// handles click of modal image
+var imageSelectSearch = function(_this) {
+  var src = $(_this).attr("src");
+
+  $("#modal").css("display", "none");
+  $(".img-preview").attr("src", src);
+  $("#results").html("");
+  $("#results").append("<div id=searching>Searching For Results...</div>");
+
+  var image = src.split('/')[2];
+  var imageName = image.split('.')[0];
+
+  $("#preview-name").text('QUERY: '+imageName);
+
+  $.ajax({
+    url: "/search",
+    data: {img: image},
+    cache: false,
+    type: 'POST',
+    success: function(data){
+      displayResults(data.results);
+    },
+    error: function(error){
+      console.log(error.toString()); 
+    }
+   });
+
+}
+
+//display results
+var displayResults = function(data){
+
+  $("#results").html("");
+
+  for(var i = 0; i < data.length; i++){
+    var image = data[i].image;
+    var score = data[i].score;
+    var element = "<div class=img-result><img class=responsive src="+imagePath+image+"/>\
+                   <div class=img-info>"+"<span class=image-name>IMAGE: "+image.split('.')[0]+"</span>\
+                   <span class=img-score>SCORE: "+score+"</span></div></div>"
+    $("#results").append(element);
+  }
+}
+
+//Controls the opening and closing of the modal
+var modalControl = function(){
+
+  // Get the modal
+  var modal = document.getElementById("modal");
+
+  // Get the button that opens the modal
+  var btn = document.getElementById("select");
+  console.log(modal);
+  
+  // Get the <span> element that closes the modal
+  var span = document.getElementsByClassName("close")[0];
+  
+  // When the user clicks the button, open the modal 
+  btn.onclick = function() {
+    modal.style.display = "block";
+  }
+  
+  // When the user clicks on <span> (x), close the modal
+  span.onclick = function() {
+    modal.style.display = "none";
+  }
+  
+  // When the user clicks anywhere outside of the modal, close it
+  window.onclick = function(event) {
+    if (event.target == modal) {
+      modal.style.display = "none";
+    }
+  }
+}

app/templates/_base.html

@@ -0,0 +1,51 @@
+<!DOCTYPE html>
+<html lang="en">
+
+<head>
+
+  <meta charset="utf-8">
+  <meta http-equiv="X-UA-Compatible" content="IE=edge">
+  <meta name="viewport" content="width=device-width, initial-scale=1">
+  <meta name="description" content="content based image retrieval system">
+  <meta name="author" content="Kene Udeh">
+
+  <title>Image Search</title>
+
+  <!-- stylesheets -->
+  <link href="{{ url_for('static', filename='main.css') }}" rel="stylesheet">
+
+  <!-- Scripts -->
+  <script src="https://ajax.googleapis.com/ajax/libs/jquery/3.4.0/jquery.min.js" type="text/javascript"></script>
+  <script src="{{ url_for('static', filename='main.js') }}" type="text/javascript"></script>
+
+</head>
+
+<body>
+
+    <!-- Page Content -->
+    <div class="container">
+
+      <!-- messages -->
+      {% for message in get_flashed_messages() %}
+        <div class="alert alert-success alert-dismissible" role="alert">
+          <button type="button" class="close" data-dismiss="alert"><span aria-hidden="true">&times;</span><span class="sr-only">Close</span></button>
+          {{ message }}
+        </div>
+      {% endfor %}
+
+      <!-- child template -->
+      {% block content %}{% endblock %}
+      {% block modal %}{% endblock %}
+      
+      <!-- errors -->
+      {% if error %}
+      <p class="error"><strong>Error:</strong> {{ error }}</p>
+      {% endif %}
+
+
+    </div>
+    <!-- /.container -->
+
+</body>
+
+</html>

app/templates/index.html

@@ -0,0 +1,38 @@
+{% extends "_base.html" %}
+
+{% block content %}
+
+<div class="header">
+  <h1 class="title">SEARCH BY IMAGE</h1>
+</div>
+<hr/>
+
+<div class="display">
+  <div class="left-pane">
+    <img src="static/init-preview.png" alt="preview image" class="img-preview">
+    <div id="preview-name">Preview</div>
+    <div id="select" class="image-select">
+      Select & Search
+    </div>
+  </div>
+  <div class="right-pane">
+    <div id="results" class="search-results">
+
+    </div>
+  </div>
+</div>
+<!-- The Modal -->
+<div id="modal" class="modal-display">
+
+    <!-- Modal content -->
+    <div class="modal-content">
+        <span class="close">&times;</span>
+        <div class="modal-images-list">
+
+        </div>
+    </div>
+      
+</div>
+
+{% endblock %}
+

venv/Lib/site-packages/_distutils_hack/__init__.py

@@ -0,0 +1,222 @@
+# don't import any costly modules
+import sys
+import os
+
+
+is_pypy = '__pypy__' in sys.builtin_module_names
+
+
+def warn_distutils_present():
+    if 'distutils' not in sys.modules:
+        return
+    if is_pypy and sys.version_info < (3, 7):
+        # PyPy for 3.6 unconditionally imports distutils, so bypass the warning
+        # https://foss.heptapod.net/pypy/pypy/-/blob/be829135bc0d758997b3566062999ee8b23872b4/lib-python/3/site.py#L250
+        return
+    import warnings
+
+    warnings.warn(
+        "Distutils was imported before Setuptools, but importing Setuptools "
+        "also replaces the `distutils` module in `sys.modules`. This may lead "
+        "to undesirable behaviors or errors. To avoid these issues, avoid "
+        "using distutils directly, ensure that setuptools is installed in the "
+        "traditional way (e.g. not an editable install), and/or make sure "
+        "that setuptools is always imported before distutils."
+    )
+
+
+def clear_distutils():
+    if 'distutils' not in sys.modules:
+        return
+    import warnings
+
+    warnings.warn("Setuptools is replacing distutils.")
+    mods = [
+        name
+        for name in sys.modules
+        if name == "distutils" or name.startswith("distutils.")
+    ]
+    for name in mods:
+        del sys.modules[name]
+
+
+def enabled():
+    """
+    Allow selection of distutils by environment variable.
+    """
+    which = os.environ.get('SETUPTOOLS_USE_DISTUTILS', 'local')
+    return which == 'local'
+
+
+def ensure_local_distutils():
+    import importlib
+
+    clear_distutils()
+
+    # With the DistutilsMetaFinder in place,
+    # perform an import to cause distutils to be
+    # loaded from setuptools._distutils. Ref #2906.
+    with shim():
+        importlib.import_module('distutils')
+
+    # check that submodules load as expected
+    core = importlib.import_module('distutils.core')
+    assert '_distutils' in core.__file__, core.__file__
+    assert 'setuptools._distutils.log' not in sys.modules
+
+
+def do_override():
+    """
+    Ensure that the local copy of distutils is preferred over stdlib.
+
+    See https://github.com/pypa/setuptools/issues/417#issuecomment-392298401
+    for more motivation.
+    """
+    if enabled():
+        warn_distutils_present()
+        ensure_local_distutils()
+
+
+class _TrivialRe:
+    def __init__(self, *patterns):
+        self._patterns = patterns
+
+    def match(self, string):
+        return all(pat in string for pat in self._patterns)
+
+
+class DistutilsMetaFinder:
+    def find_spec(self, fullname, path, target=None):
+        # optimization: only consider top level modules and those
+        # found in the CPython test suite.
+        if path is not None and not fullname.startswith('test.'):
+            return
+
+        method_name = 'spec_for_{fullname}'.format(**locals())
+        method = getattr(self, method_name, lambda: None)
+        return method()
+
+    def spec_for_distutils(self):
+        if self.is_cpython():
+            return
+
+        import importlib
+        import importlib.abc
+        import importlib.util
+
+        try:
+            mod = importlib.import_module('setuptools._distutils')
+        except Exception:
+            # There are a couple of cases where setuptools._distutils
+            # may not be present:
+            # - An older Setuptools without a local distutils is
+            #   taking precedence. Ref #2957.
+            # - Path manipulation during sitecustomize removes
+            #   setuptools from the path but only after the hook
+            #   has been loaded. Ref #2980.
+            # In either case, fall back to stdlib behavior.
+            return
+
+        class DistutilsLoader(importlib.abc.Loader):
+            def create_module(self, spec):
+                mod.__name__ = 'distutils'
+                return mod
+
+            def exec_module(self, module):
+                pass
+
+        return importlib.util.spec_from_loader(
+            'distutils', DistutilsLoader(), origin=mod.__file__
+        )
+
+    @staticmethod
+    def is_cpython():
+        """
+        Suppress supplying distutils for CPython (build and tests).
+        Ref #2965 and #3007.
+        """
+        return os.path.isfile('pybuilddir.txt')
+
+    def spec_for_pip(self):
+        """
+        Ensure stdlib distutils when running under pip.
+        See pypa/pip#8761 for rationale.
+        """
+        if self.pip_imported_during_build():
+            return
+        clear_distutils()
+        self.spec_for_distutils = lambda: None
+
+    @classmethod
+    def pip_imported_during_build(cls):
+        """
+        Detect if pip is being imported in a build script. Ref #2355.
+        """
+        import traceback
+
+        return any(
+            cls.frame_file_is_setup(frame) for frame, line in traceback.walk_stack(None)
+        )
+
+    @staticmethod
+    def frame_file_is_setup(frame):
+        """
+        Return True if the indicated frame suggests a setup.py file.
+        """
+        # some frames may not have __file__ (#2940)
+        return frame.f_globals.get('__file__', '').endswith('setup.py')
+
+    def spec_for_sensitive_tests(self):
+        """
+        Ensure stdlib distutils when running select tests under CPython.
+
+        python/cpython#91169
+        """
+        clear_distutils()
+        self.spec_for_distutils = lambda: None
+
+    sensitive_tests = (
+        [
+            'test.test_distutils',
+            'test.test_peg_generator',
+            'test.test_importlib',
+        ]
+        if sys.version_info < (3, 10)
+        else [
+            'test.test_distutils',
+        ]
+    )
+
+
+for name in DistutilsMetaFinder.sensitive_tests:
+    setattr(
+        DistutilsMetaFinder,
+        f'spec_for_{name}',
+        DistutilsMetaFinder.spec_for_sensitive_tests,
+    )
+
+
+DISTUTILS_FINDER = DistutilsMetaFinder()
+
+
+def add_shim():
+    DISTUTILS_FINDER in sys.meta_path or insert_shim()
+
+
+class shim:
+    def __enter__(self):
+        insert_shim()
+
+    def __exit__(self, exc, value, tb):
+        remove_shim()
+
+
+def insert_shim():
+    sys.meta_path.insert(0, DISTUTILS_FINDER)
+
+
+def remove_shim():
+    try:
+        sys.meta_path.remove(DISTUTILS_FINDER)
+    except ValueError:
+        pass

venv/Lib/site-packages/_distutils_hack/override.py

@@ -0,0 +1 @@
+__import__('_distutils_hack').do_override()

venv/Lib/site-packages/distutils-precedence.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2638ce9e2500e572a5e0de7faed6661eb569d1b696fcba07b0dd223da5f5d224
+size 151