mirror of
https://github.com/geekwenjie/SmartJavaAI.git
synced 2026-09-15 14:33:03 +00:00
1、人脸模块:新增小视科技(MiniVision)活体检测模型
2、人脸模块:新增阿里通义工作室活体检测模型 3、人脸模块:新增2个表情识别模型 4、人脸模块:新增InsightFace、ElasticFace人脸识别模型 5、人脸模块:新增Seetaface6质量评估模型 6、目标检测模块:开放更多自定义模型参数 7、人脸模块:支持base64图片 8、实现接口 AutoCloseable,支持资源的自动释放 9、OCR模块:解决加方向矫正后无法连续识别bug 10、人脸模块:解决人脸更新后缓存问题 11、优化部分功能
This commit is contained in:
@@ -0,0 +1,100 @@
|
||||
package smartai.examples.nlp.translation;
|
||||
|
||||
import ai.djl.util.JsonUtils;
|
||||
import cn.smartjavaai.common.entity.R;
|
||||
import cn.smartjavaai.common.enums.DeviceEnum;
|
||||
import cn.smartjavaai.translation.config.TranslationModelConfig;
|
||||
import cn.smartjavaai.translation.entity.TranslateParam;
|
||||
import cn.smartjavaai.translation.enums.LanguageCode;
|
||||
import cn.smartjavaai.translation.enums.TranslationModeEnum;
|
||||
import cn.smartjavaai.translation.factory.TranslationModelFactory;
|
||||
import cn.smartjavaai.translation.model.TranslationModel;
|
||||
import lombok.Data;
|
||||
import lombok.extern.slf4j.Slf4j;
|
||||
import org.junit.Test;
|
||||
|
||||
/**
|
||||
* 翻译Demo
|
||||
* 支持 Meta AI 开源的 NLLB-200 模型,实现 200 多种语言之间的高质量互译。
|
||||
* NLLB-200官网地址:https://github.com/facebookresearch/fairseq/tree/nllb
|
||||
* 模型下载地址:https://pan.baidu.com/s/1wf7btnb4cyBFv7DB7baHnw?pwd=1234 提取码: 1234
|
||||
* @author dwj
|
||||
*/
|
||||
@Slf4j
|
||||
public class TranslationDemo {
|
||||
|
||||
|
||||
/**
|
||||
* 翻译
|
||||
*/
|
||||
@Test
|
||||
public void translate() {
|
||||
try {
|
||||
TranslationModelConfig config = new TranslationModelConfig();
|
||||
//指定翻译模型:NLLB
|
||||
config.setModelEnum(TranslationModeEnum.NLLB_MODEL);
|
||||
//指定模型路径,需将模型路径修改为本地的模型路径
|
||||
config.setModelPath("/Users/xxx/Documents/develop/model/trans/traced_translation_cpu.pt");
|
||||
TranslationModel translationModel = TranslationModelFactory.getInstance().getModel(config);
|
||||
//翻译参数
|
||||
TranslateParam translateParam = new TranslateParam();
|
||||
//输入文字
|
||||
translateParam.setInput("你好,欢迎使用SmartJavaAI!");
|
||||
//源语言:中文
|
||||
translateParam.setSourceLanguage(LanguageCode.ZHO_HANS);
|
||||
//目标语言:英文
|
||||
translateParam.setTargetLanguage(LanguageCode.ENG_LATN);
|
||||
R<String> result = translationModel.translate(translateParam);
|
||||
if(result.isSuccess()){
|
||||
log.info("翻译结果:{}", result.getData());
|
||||
}else{
|
||||
log.error("翻译失败:{}", result.getMessage());
|
||||
}
|
||||
//目标语言:韩语
|
||||
translateParam.setTargetLanguage(LanguageCode.KOR_HANG);
|
||||
R<String> result2 = translationModel.translate(translateParam);
|
||||
if(result2.isSuccess()){
|
||||
log.info("翻译结果:{}", result2.getData());
|
||||
}else{
|
||||
log.error("翻译失败:{}", result2.getMessage());
|
||||
}
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* GPU 翻译
|
||||
*/
|
||||
@Test
|
||||
public void translateGpu() {
|
||||
try {
|
||||
TranslationModelConfig config = new TranslationModelConfig();
|
||||
//指定翻译模型:NLLB
|
||||
config.setModelEnum(TranslationModeEnum.NLLB_MODEL);
|
||||
//指定设备:GPU
|
||||
config.setDevice(DeviceEnum.GPU);
|
||||
//指定模型路径,需将模型路径修改为本地的 GPU 模型路径
|
||||
config.setModelPath("/Users/xxx/Documents/develop/model/trans/traced_translation_gpu.pt");
|
||||
//获取翻译模型
|
||||
TranslationModel translationModel = TranslationModelFactory.getInstance().getModel(config);
|
||||
//翻译参数
|
||||
TranslateParam translateParam = new TranslateParam();
|
||||
//输入文字
|
||||
translateParam.setInput("你好,欢迎使用SmartJavaAI!");
|
||||
//源语言:中文
|
||||
translateParam.setSourceLanguage(LanguageCode.ZHO_HANS);
|
||||
//目标语言:韩语
|
||||
translateParam.setTargetLanguage(LanguageCode.ENG_LATN);
|
||||
R<String> result = translationModel.translate(translateParam);
|
||||
if(result.isSuccess()){
|
||||
log.info("翻译结果:{}", result.getData());
|
||||
}else{
|
||||
log.error("翻译失败:{}", result.getMessage());
|
||||
}
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
@@ -0,0 +1,3 @@
|
||||
Manifest-Version: 1.0
|
||||
Main-Class: smartai.examples.face.SeetaFace6LinuxDemo
|
||||
|
||||
14
examples/translation-example/src/main/resources/logback.xml
Normal file
14
examples/translation-example/src/main/resources/logback.xml
Normal file
@@ -0,0 +1,14 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<!-- 步骤2: 配置文件 (src/main/resources/logback.xml) -->
|
||||
<configuration scan="true" scanPeriod="30 seconds">
|
||||
<!-- 控制台日志输出 -->
|
||||
<appender name="CONSOLE" class="ch.qos.logback.core.ConsoleAppender">
|
||||
<encoder>
|
||||
<pattern>%d{yyyy-MM-dd HH:mm:ss.SSS} [%thread] %highlight(%-5level) %cyan(%logger{36}) - %msg%n</pattern>
|
||||
</encoder>
|
||||
</appender>
|
||||
|
||||
<root level="DEBUG">
|
||||
<appender-ref ref="CONSOLE" />
|
||||
</root>
|
||||
</configuration>
|
||||
Reference in New Issue
Block a user