coffee-burns-snake/OpenImajTest.java at main · FHOOEAIST/coffee-burns-snake · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
/*
 * Copyright (c) 2021 the original author or authors.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */

import org.openimaj.image.FImage;
import org.openimaj.image.processing.convolution.FourierConvolve;

public class OpenImajTest {
    public static int current = 0;

    public static void main(String[] args) {
        int warmups = 100;
        int runs = 1000;
        int imageWidth = 1920;
        int imageHeight = 1080;
        int kernelSize = 5;

        FImage image = new FImage(imageWidth, imageHeight);
        for(int x = 0; x < imageWidth; x++){
            for (int y = 0; y < imageHeight; y++) {
                float cval = (float)current / 255.0f;
                image.setPixel(x, y, cval);
                current++;
                if (current > 255) current = 0;
            }
        }

        float[][] mask = new float[][]{
                new float[]{1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize))},
                new float[]{1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize))},
                new float[]{1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize))},
                new float[]{1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize))},
                new float[]{1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize)), 1.0f / ((float) (kernelSize * kernelSize))},
        };

        for (int i = 0; i < warmups; i++) {
            FourierConvolve.convolve(image, mask, false);
        }

        for (int i = 0; i < runs; i++) {
            long before = System.currentTimeMillis();
            FourierConvolve.convolve(image, mask, false);
            long after = System.currentTimeMillis();
            System.out.println("" + (after - before));
        }

    }
}