forked from liferay/liferay-portal
-
Notifications
You must be signed in to change notification settings - Fork 0
/
GetLabelProbabilitiesProcessCallable.java
39 lines (31 loc) · 1.19 KB
/
GetLabelProbabilitiesProcessCallable.java
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
/**
* Copyright (c) 2000-present Liferay, Inc. All rights reserved.
*
* This library is free software; you can redistribute it and/or modify it under
* the terms of the GNU Lesser General Public License as published by the Free
* Software Foundation; either version 2.1 of the License, or (at your option)
* any later version.
*
* This library is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
* FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
* details.
*/
package com.liferay.document.library.asset.auto.tagger.tensorflow.internal.process;
import com.liferay.petra.process.ProcessCallable;
import com.liferay.petra.process.ProcessException;
/**
* @author Shuyang Zhou
*/
public class GetLabelProbabilitiesProcessCallable
implements ProcessCallable<float[]> {
public GetLabelProbabilitiesProcessCallable(byte[] imageBytes) {
_imageBytes = imageBytes;
}
@Override
public float[] call() throws ProcessException {
return InceptionImageLabelerUtil.getLabelProbabilities(_imageBytes);
}
private static final long serialVersionUID = 1L;
private final byte[] _imageBytes;
}