feat(ai): 集成动漫人物分割与面部解析AI模型- 添加 DJL 深度学习框架依赖项以支持 PyTorch 和 ONNX Runtime 引擎
- 实现 Anime2VividModelWrapper 封装类用于动漫人物前景背景分离 - 开发 AnimeModelWrapper用于精细的动漫面部特征(如头发、眼睛)分割 - 创建配套的标签调色板和结果处理工具类提升可视化效果 - 增加多个测试用例验证不同AI模型的推理及文件输出功能 - 支持通过 synset.txt 自定义模型标签并增强命令行可测试性
This commit is contained in:
10
build.gradle
10
build.gradle
@@ -51,6 +51,16 @@ dependencies {
|
||||
implementation files('libs/dog api 1.3.jar')
|
||||
implementation files('libs/DesktopWallpaperSdk-1.0-SNAPSHOT.jar')
|
||||
|
||||
// === DJL API ===
|
||||
implementation platform('ai.djl:bom:0.35.0')
|
||||
implementation 'ai.djl:api'
|
||||
implementation 'ai.djl:model-zoo'
|
||||
implementation 'ai.djl.pytorch:pytorch-model-zoo:0.35.0'
|
||||
implementation 'ai.djl.pytorch:pytorch-engine'
|
||||
implementation 'ai.djl:basicdataset'
|
||||
implementation 'ai.djl.onnxruntime:onnxruntime-engine'
|
||||
runtimeOnly 'ai.djl.pytorch:pytorch-native-cpu:2.7.1'
|
||||
runtimeOnly 'ai.djl.onnxruntime:onnxruntime-native-cpu:1.3.0'
|
||||
// === 核心工具库 ===
|
||||
implementation 'com.google.code.gson:gson:2.10.1' // 统一版本
|
||||
implementation 'org.apache.logging.log4j:log4j-api:2.20.0'
|
||||
|
||||
Reference in New Issue
Block a user