package com.lanbao;
|
|
import ai.djl.modality.Classifications;
|
import org.jeecg.modules.dry.util.HerbUtil;
|
import org.junit.Test;
|
|
import java.io.File;
|
import java.io.FileInputStream;
|
import java.io.FileNotFoundException;
|
import java.io.InputStream;
|
import java.util.List;
|
|
public class AiTest {
|
@Test
|
public void test() {
|
try {
|
String img = "E:\\testImg\\szp1.jpg";
|
HerbUtil herbUtil = new HerbUtil();
|
File file = new File(img);
|
InputStream is = new FileInputStream(file);
|
List<Classifications.Classification> predict = herbUtil.predict(is);
|
System.out.println(predict);
|
} catch (FileNotFoundException e) {
|
throw new RuntimeException(e);
|
}
|
}
|
}
|
