jenkins-bot has submitted this change and it was merged.
Change subject: Porting match_images.py from compat to core/scripts
......................................................................
Porting match_images.py from compat to core/scripts
Added match_images.py in core/scripts/ as part of Pywikibot:Compat
to Core migration making it compatible with pywikibot module. Needed to
make use of http from pywikibot.comms for fetching URLs.
Bug T66871
Change-Id: I7aa6cc3e1c97e32d50621418cb9903f35321157f
---
A scripts/match_images.py
1 file changed, 191 insertions(+), 0 deletions(-)
Approvals:
Ladsgroup: Looks good to me, approved
jenkins-bot: Verified
diff --git a/scripts/match_images.py b/scripts/match_images.py
new file mode 100644
index 0000000..9a8204c
--- /dev/null
+++ b/scripts/match_images.py
@@ -0,0 +1,191 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+"""
+Program to match two images based on histograms.
+
+Usage:
+match_images.py ImageA ImageB
+It is essential to provide two images to work on.
+example. - match_images.py ImageA.jpg ImageB.jpg
+
+¶ms;
+
+Furthermore, the following command line parameters are supported:
+
+-otherfamily Mentioned family with this parameter will be preferred for
+ fetching file usage details instead of the default
+ family retrieved from user-congig.py script.
+
+-otherlang Mentioned lang with this parameter will be preferred for
+ fetching file usage details instead of the default
+ mylang retrieved from user-congig.py script.
+
+This is just a first version so that other people can play around with it.
+Expect the code to change a lot!
+"""
+#
+# (c) Multichill, 2009
+# (c) pywikibot team, 2009-2015
+#
+# Distributed under the terms of the MIT license.
+#
+from __future__ import division
+__version__ = '$Id$'
+
+
+import io
+from PIL import Image
+
+import pywikibot
+from pywikibot.comms import http
+
+
+def match_image_pages(imagePageA, imagePageB):
+ """The function expects two image page objects.
+
+ It will return True if the image are the same and False if the images are
+ not the same
+
+ """
+ imageA = get_image_from_image_page(imagePageA)
+ imageB = get_image_from_image_page(imagePageB)
+
+ (imA_width, imA_height) = imageA.size
+ (imB_width, imB_height) = imageB.size
+
+ imageB = imageB.resize((imA_width, imA_height))
+
+ imageA_topleft = imageA.crop((0, 0, imA_width // 2, imA_height // 2))
+ imageB_topleft = imageB.crop((0, 0, imA_width // 2, imA_height // 2))
+
+ imageA_topright = imageA.crop((imA_width // 2, 0, imA_width,
+ imA_height // 2))
+ imageB_topright = imageB.crop((imA_width // 2, 0, imA_width,
+ imA_height // 2))
+
+ imageA_bottomleft = imageA.crop((0, imA_height // 2, imA_width // 2,
+ imA_height))
+ imageB_bottomleft = imageB.crop((0, imA_height // 2, imA_width // 2,
+ imA_height))
+
+ imageA_bottomright = imageA.crop((imA_width // 2, imA_height // 2,
+ imA_width, imA_height))
+ imageB_bottomright = imageB.crop((imA_width // 2, imA_height // 2,
+ imA_width, imA_height))
+
+ imageA_center = imageA.crop((int(imA_width * 0.25), int(imA_height * 0.25),
+ int(imA_width * 0.75), int(imA_height * 0.75)))
+ imageB_center = imageB.crop((int(imA_width * 0.25), int(imA_height * 0.25),
+ int(imA_width * 0.75), int(imA_height * 0.75)))
+
+ wholeScore = match_images(imageA, imageB)
+ topleftScore = match_images(imageA_topleft, imageB_topleft)
+ toprightScore = match_images(imageA_topright, imageB_topright)
+ bottomleftScore = match_images(imageA_bottomleft, imageB_bottomleft)
+ bottomrightScore = match_images(imageA_bottomright, imageB_bottomright)
+ centerScore = match_images(imageA_center, imageB_center)
+ averageScore = (wholeScore + topleftScore + toprightScore +
+ bottomleftScore + bottomrightScore + centerScore) / 6
+
+ print (u'Whole image ' + str(wholeScore))
+ print (u'Top left of image ' + str(topleftScore))
+ print (u'Top right of image ' + str(toprightScore))
+ print (u'Bottom left of image ' + str(bottomleftScore))
+ print (u'Bottom right of image ' + str(bottomrightScore))
+ print (u'Center of image ' + str(centerScore))
+ print (u' -------------')
+ print (u'Average ' + str(averageScore))
+
+ # Hard coded at 80%, change this later on.
+ if averageScore > 0.8:
+ print (u'We have a match!')
+ return True
+ else:
+ print (u'Not the same.')
+ return False
+
+
+def get_image_from_image_page(imagePage):
+ """ Get the image object to work based on an imagePage object. """
+ imageBuffer = None
+ imageURL = imagePage.fileUrl()
+ imageURLopener = http.fetch(imageURL)
+ imageBuffer = io.BytesIO(imageURLopener.raw[:])
+ image = Image.open(imageBuffer)
+ return image
+
+
+def match_images(imageA, imageB):
+ """ Match two image objects. Return the ratio of pixels that match. """
+ histogramA = imageA.histogram()
+ histogramB = imageB.histogram()
+
+ totalMatch = 0
+ totalPixels = 0
+
+ if len(histogramA) != len(histogramB):
+ return 0
+
+ for i in range(0, len(histogramA)):
+ totalMatch = totalMatch + min(histogramA[i], histogramB[i])
+ totalPixels = totalPixels + max(histogramA[i], histogramB[i])
+
+ if totalPixels == 0:
+ return 0
+
+ return totalMatch / totalPixels * 100
+
+
+def main(*args):
+ """ Extracting file page information of images to work on and initiate
matching. """
+ images = []
+ other_family = u''
+ other_lang = u''
+ imagePageA = None
+ imagePageB = None
+
+ # Read commandline parameters.
+ local_args = pywikibot.handle_args(args)
+
+ for arg in local_args:
+ if arg.startswith('-otherfamily:'):
+ if len(arg) == len('-otherfamily:'):
+ other_family = pywikibot.input(u'What family do you want to
use?')
+ else:
+ other_family = arg[len('-otherfamily:'):]
+ elif arg.startswith('-otherlang:'):
+ if len(arg) == len('-otherlang:'):
+ other_lang = pywikibot.input(u'What language do you want to
use?')
+ else:
+ other_lang = arg[len('otherlang:'):]
+ else:
+ images.append(arg)
+
+ if len(images) != 2:
+ pywikibot.showHelp('match_images')
+ pywikibot.error('Require two images to work on.')
+ return
+
+ else:
+ pass
+
+ imagePageA = pywikibot.page.FilePage(pywikibot.Site(),
+ images[0])
+ if other_lang:
+ if other_family:
+ imagePageB = pywikibot.page.FilePage(pywikibot.Site(
+ other_lang, other_family),
+ images[1])
+ else:
+ imagePageB = pywikibot.page.FilePage(pywikibot.Site(
+ other_lang),
+ images[1])
+ else:
+ imagePageB = pywikibot.page.FilePage(pywikibot.Site(),
+ images[1])
+
+ match_image_pages(imagePageA, imagePageB)
+
+
+if __name__ == "__main__":
+ main()
--
To view, visit https://gerrit.wikimedia.org/r/181988
To unsubscribe, visit https://gerrit.wikimedia.org/r/settings
Gerrit-MessageType: merged
Gerrit-Change-Id: I7aa6cc3e1c97e32d50621418cb9903f35321157f
Gerrit-PatchSet: 9
Gerrit-Project: pywikibot/core
Gerrit-Branch: master
Gerrit-Owner: Prianka <[email protected]>
Gerrit-Reviewer: John Vandenberg <[email protected]>
Gerrit-Reviewer: Ladsgroup <[email protected]>
Gerrit-Reviewer: Merlijn van Deen <[email protected]>
Gerrit-Reviewer: Prianka <[email protected]>
Gerrit-Reviewer: XZise <[email protected]>
Gerrit-Reviewer: jenkins-bot <>
_______________________________________________
MediaWiki-commits mailing list
[email protected]
https://lists.wikimedia.org/mailman/listinfo/mediawiki-commits
