JAVA调用SKIL模型输出多个INDArrays

1、Main.java

package ai.skymind.demo;

import ai.skymind.ApiClient;
import ai.skymind.ApiException;
import ai.skymind.Configuration;
import ai.skymind.skil.DefaultApi;
import ai.skymind.skil.model.LoginRequest;
import ai.skymind.skil.model.LoginResponse;
import ai.skymind.skil.model.MultiPredictResponse;
import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.serde.base64.Nd4jBase64;

import java.io.File;
import java.io.IOException;
import java.util.UUID;

public class Main {

    public static void main(String[] args) throws ApiException, IOException {
        String imagePath = args[0];
        File imageFile = new File(imagePath);

        if (!imageFile.exists()) {
            System.out.println("Specified file did not exist.");
            System.exit(1);
        }
        ApiClient defaultClient = Configuration.getDefaultApiClient();
        defaultClient.setBasePath("http://127.0.0.1:9008");

        // Configure API key authorization: api_key
        DefaultApi skil = new DefaultApi();
        LoginRequest loginRequest = new LoginRequest();
        loginRequest.setUserId("admin");
        loginRequest.setPassword("admin123");
        LoginResponse loginResponse = skil.login(loginRequest);

        String token = loginResponse.getToken();

        defaultClient.setApiKey(token);
        defaultClient.setApiKeyPrefix("Bearer");

        MultiPredictResponse multiPredictResponse = skil.multipredictimage(
                imageFile,
                UUID.randomUUID().toString(),
                false,
                "demo",
                "default",
                "wideresnet"
        );

        assert(multiPredictResponse.getOutputs().size() == 2);
        INDArray array1 = Nd4jBase64.fromBase64(multiPredictResponse.getOutputs().get(0).getArray());
        INDArray array2 = Nd4jBase64.fromBase64(multiPredictResponse.getOutputs().get(1).getArray());

        System.out.println(array1.toString());
        System.out.println(array2.toString());
    }
}

2、pom.xml

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>

    <groupId>ai.skymind.demo</groupId>
    <artifactId>skil-clients-sample</artifactId>
    <version>1.0-SNAPSHOT</version>

    <properties>
        <maven.compiler.target>1.8</maven.compiler.target>
        <maven.compiler.source>1.8</maven.compiler.source>
    </properties>

    <dependencies>
        <dependency>
            <groupId>ai.skymind</groupId>
            <artifactId>skil-client</artifactId>
            <version>1.2.1-SNAPSHOT</version>
            <scope>compile</scope>
        </dependency>

        <dependency>
            <groupId>org.nd4j</groupId>
            <artifactId>nd4j-native-platform</artifactId>
            <version>1.0.0-beta3</version>
        </dependency>
    </dependencies>
</project>

3、git https://github.com/wmeddie/skil-clients-sample.git

           https://github.com/wmeddie/skil-clients-sample/blob/master/src/main/java/ai/skymind/demo/Main.java#L45

4、可以把Main.class拷贝到 skil-clinets-master  中进行测试

 

5、python的案例

 

转载于:https://my.oschina.net/u/3209854/blog/3063217