Ejemplo n.º 1
0
def main():
    np.set_printoptions(linewidth=150)

    # Read the code for this file and hash it
    with open('sprite_tree.py', 'r') as fp:
        my_hash_value = sha1(fp.read()).hexdigest()

    # The pickling should really be done in the sprite_tree, but this is less
    # of a PITA
    file_name = 'pickled_sprites/{}'.format(my_hash_value)
    sprites = {k: reduce_image(v) for k, v in get_sprites('sprites').items()}
    if os.path.exists(file_name):
        cprint('Unpickling SpriteTree', 'green')
        with open(file_name, 'r') as fp:
            st = pickle.load(fp)
    else:
        cprint('Pickling SpriteTree', 'yellow')
        st = SpriteTree(sprites)
        with open(file_name, 'w') as fp:
            pickle.dump(st, fp)

    img_f = ImageFactory(sprites, background_color=(255, 132, 138))

    image = np.zeros((224, 256, 3), dtype='ubyte')
    for row in image:
        for px in row:
            np.copyto(px, img_f._bg_color)

    s = sprites['small_jump']
    # blit(self, image, sprite, mask, x, y):
    image = img_f.blit(image, s[:, :, :3], s[:, :, 3], 50, 50)
    si = SegmentedImage(st, image)
    print(np.nonzero(np.where(si._hashed == si.background_color(), 0, si._hashed)))
    bounding_boxes, names = si.find_sprites()
    cprint(bounding_boxes, 'cyan')
    cprint(names, 'blue')
    sys.exit()

    file_names = glob('data/*')
    images = []
    for fn in range(len(file_names))[1224:3224]:
        fn = 'data/{}.png'.format(fn)
        original = cv2.imread(fn, cv2.IMREAD_COLOR)  # [300:, :]
        image = reduce_image(original)
        images.append(image)

    start = time.time()
    for i, image in enumerate(images):
        si = SegmentedImage(st, image)
        si.find_sprites()
        cprint((time.time() - start) / float(i + 1), 'green')

    cprint((time.time() - start) / 1000.0, 'green')
def main():
    sprites = {k: reduce_image(v) for k, v in get_sprites('sprites').items()}
    #sprites = {
    #    'small_jump': sprites['small_jump'],
    #    'big_run_1': sprites['big_run_1'],
    #    'big_run_3': sprites['big_run_3'],
    #}
    st = SpriteTree(sprites)

    probs = st.get_horizontal_probability(3, 3)
    named_probs = sorted(zip(st._names, probs), key=lambda a: a[1])

    for n, p in named_probs:
        print('{}: {}'.format(n, p))
    print
def main():
    sprites = {k: reduce_image(v) for k, v in get_sprites('sprites').items()}
    #sprites = {
    #    'small_jump': sprites['small_jump'],
    #    'big_run_1': sprites['big_run_1'],
    #    'big_run_3': sprites['big_run_3'],
    #}
    st = SpriteTree(sprites)

    probs = st.get_horizontal_probability(3, 3)
    named_probs = sorted(zip(st._names, probs), key=lambda a: a[1])

    for n, p in named_probs:
        print('{}: {}'.format(n, p))
    print
def main():
    sprites = {k: reduce_image(v) for k, v in get_sprites('sprites').items()}

    img_factory = ImageFactory(sprites, background_color=(255, 132, 138))
    blitted = img_factory.new_image(num_sprites=10)
    show_image(blitted['image'])
def main():
    sprites = {k: reduce_image(v) for k, v in get_sprites('sprites').items()}

    img_factory = ImageFactory(sprites, background_color=(255, 132, 138))
    blitted = img_factory.new_image(num_sprites=10)
    show_image(blitted['image'])