干燥机配套车间生产管理系统/云平台服务端
bsw215583320
2024-04-09 ae2d52827d289dd985014a63c08b0913d083b215
jeecg-module-dry/jeecg-module-dry-start/src/main/test/java/com/lanbao/AiTest.java
@@ -1,27 +1,27 @@
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);
        }
    }
}
//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);
//        }
//    }
//}