fixed_image_sampler.py

#!/usr/bin/python3

# Only needed if not installed system wide
import sys
sys.path.insert(0, '../..')


# Program start here
#
# Load images/all_shapes.png and images/shape_blue_circle.png
# as a part of haystack and needle, then find the needle in
# the haystack, and dump the results of the matching in a
# tmp folder in examples. The main purpose of this sample is
# to be reused as a tool for matching fixed needle/haystack
# pairs in order to figure out the best parameter configuration
# for successful matching.


import logging
import shutil

from guibot.config import GlobalConfig
from guibot.imagelogger import ImageLogger
from guibot.fileresolver import FileResolver
from guibot.errors import *
from guibot.target import *
from guibot.finder import *


# Parameters to toy with
file_resolver = FileResolver()
file_resolver.add_path('images/')
BACKEND = "template"
# could be Text('Text') or any other target type
NEEDLE = Image('shape_blue_circle')
HAYSTACK = Image('all_shapes')
# image logging variables
LOGPATH = './tmp/'
REMOVE_LOGPATH = False


# Overall logging setup
handler = logging.StreamHandler()
logging.getLogger('').addHandler(handler)
logging.getLogger('').setLevel(logging.DEBUG)
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
handler.setFormatter(formatter)
GlobalConfig.image_logging_level = 0
GlobalConfig.image_logging_destination = LOGPATH
GlobalConfig.image_logging_step_width = 4
ImageLogger.step = 1


# Main configuration steps
GlobalConfig.find_backend = BACKEND
if GlobalConfig.find_backend == "autopy":
    finder = AutoPyFinder(synchronize=False)
elif GlobalConfig.find_backend == "contour":
    finder = ContourFinder(synchronize=False)
elif GlobalConfig.find_backend == "template":
    finder = TemplateFinder(synchronize=False)
elif GlobalConfig.find_backend == "feature":
    finder = FeatureFinder(synchronize=False)
elif GlobalConfig.find_backend == "cascade":
    finder = CascadeFinder(synchronize=False)
elif GlobalConfig.find_backend == "text":
    finder = TextFinder(synchronize=False)
elif GlobalConfig.find_backend == "tempfeat":
    finder = TemplateFeatureFinder(synchronize=False)
elif GlobalConfig.find_backend == "deep":
    finder = DeepFinder(synchronize=False)
# example configuration from various finder types
#finder.configure(text_detector="contours")
#finder.configure_backend(backend="sqdiff_normed", category="template")
#finder.params["find"]["similarity"].value = 0.7
#finder.params["tempfeat"]["front_similarity"].value = 0.5
#finder.params["feature"]["ransacReprojThreshold"].value = 25.0
#finder.params["fdetect"]["MaxFeatures"].value = 10
#finder.params["text"]["datapath"].value = "../../misc"
#finder.params["ocr"]["oem"].value = 0
#finder.params["tdetect"]["verticalVariance"].value = 5
#finder.params["threshold"]["blockSize"].value = 3
# synchronize at this stage to take into account all configuration
finder.synchronize()


# Main matching step
finder.find(NEEDLE, HAYSTACK)


# Final cleanup steps
if REMOVE_LOGPATH:
    shutil.rmtree(LOGPATH)
GlobalConfig.image_logging_level = logging.ERROR
GlobalConfig.image_logging_destination = "./imglog"
GlobalConfig.image_logging_step_width = 3