| | |
| | | import ai.djl.modality.cv.transform.*; |
| | | import ai.djl.modality.cv.translator.ImageClassificationTranslator; |
| | | import ai.djl.translate.Translator; |
| | | import lombok.extern.slf4j.Slf4j; |
| | | import org.springframework.core.io.Resource; |
| | | import org.springframework.core.io.support.PathMatchingResourcePatternResolver; |
| | | import org.springframework.core.io.support.ResourcePatternResolver; |
| | | import org.springframework.stereotype.Component; |
| | | |
| | | import javax.imageio.ImageIO; |
| | |
| | | import java.util.ArrayList; |
| | | import java.util.List; |
| | | |
| | | @Slf4j |
| | | @Component |
| | | public class HerbUtil { |
| | | |
| | |
| | | System.out.println(output); |
| | | result = output.topK(); |
| | | } catch (Exception e) { |
| | | log.error("药材识别异常!!"); |
| | | log.error(input.toString()); |
| | | log.error(predictor.toString()); |
| | | e.printStackTrace(); |
| | | } |
| | | return result; |
| | |
| | | //随便起名 |
| | | Model model = Model.newInstance("model", Device.cpu()); |
| | | try { |
| | | InputStream inputStream = HerbUtil.class.getClassLoader().getResourceAsStream("model34.pt"); |
| | | // ResourcePatternResolver resolver = new PathMatchingResourcePatternResolver(); |
| | | // Resource[] resources = resolver.getResources("../pytorch/model34.pt"); |
| | | // Resource resource = resources[0]; |
| | | File f = new File("../pytorch/model34.pt"); |
| | | |
| | | InputStream inputStream = new FileInputStream(f); |
| | | // InputStream inputStream = HerbUtil.class.getClassLoader().getResourceAsStream("model34.pt"); |
| | | if (inputStream == null) { |
| | | throw new RuntimeException("找不到模型文件"); |
| | | } |