- 下载Release中的.jar和.zip文件
- 将.jar文件Add as library或利用Maven引入工程中
- 解压.zip并将解压后的weights文件夹拷贝至工程目录下(示例代码中weights文件夹存放于resources目录下)
@Test
public void testPredict(){
// 引入DetectServiceImpl类
DetectService service=new DetectServiceImpl();
// 准备传入参数
UtilInputs input=new UtilInputs();
input.setClassName("QRCode.names");
input.setModelName("QRCode.torchscript");
// 获取模型所在目录路径
input.setPath(Main.class.getResource("/weights").getPath());
input.setHeight(640);
input.setWidth(640);
// 开始预测
service.detect(null,input);
}
- 参考MacauQRCodeWeb文件夹内SpringBoot项目
- 请注意${ }表示需要根据实际情况对此处代码作出修改
- project
- src
- domain
- Predict.java
- UtilInputs.java
- Bound.java
- service
- DetectService
- impl
- DetectServiceImpl
- impl
- DetectService
- utils
- InitializeUtil.java
- GetBoundUtil.java
- domain
- target
- weights
- QRCode.names
- QRCode.torchscript
- weights.zip
- MacauQRCodeWeb
- src
若使用其他模型可能需要对源代码做出修改