Overview

[Project] [Paper] [Online Demo] [API] [Related Work: GP-GAN (for Image Blending)]

source	step 1	step 2	step 3	step 4	step 5	output

A2-RL (aka. Aesthetics Aware Reinforcement Learning) is the author's implementation of the RL-based automatic image cropping algorithm described in:

A2-RL: Aesthetics Aware Reinforcement Learning for Automatic Image Cropping   
Debang Li, Huikai Wu, Junge Zhang, Kaiqi Huang

Given a source image, our algorithm could take actions step by step to find almost the best cropping window on source image.

Usage

Input

URL

input = {
    'type': 'U',
    'url': image_path
}

Data API

input = {
    'type': 'D',
    'data': data_api_uri
}

Client Users [Extra Method]

For client users, the input is a list of bytearray reading from image files.

input = bytearray(open('xxx.jpg', 'rb').read())
result = algo.pipe([input]).result

Web Users [Extra Method]

For web (javascript) users, the input is a dict with 'image' as the keys and base64 string of the images as the values.

...
<input id="file-input" type="file" name="name" style="display: none;"/>
...

document.getElementById('file-input').onchange = function (evt) {
        var tgt = evt.target || window.event.srcElement;
        files = tgt.files;
        // FileReader support
        if (FileReader && files && files.length) {
            var reader = new FileReader();
            reader.onload = function () {
                Algorithmia.client("xxxxxxxxxxxxxxxx")
                           .algo("algo://wuhuikai/A2RL")
                           .pipe({'type': 'B', 'image': reader.result})
                           .then(function(output) {
                                ...
                           });
            }
            reader.readAsDataURL(files[0]);
        } else {
            // Not supported
            ...
        }

Output [Normalized into (0~1)]

output.result = [[xmin, ymin, xmax, ymax], [xmin, ymin, xmax, ymax], ..., [xmin, ymin, xmax, ymax]]

Examples

Source	A2-RL

Sign In

A2RL: Auto Image Cropping

Run an Example

Install & Use