You've already forked FrameTour-BE
feat(image): 添加图像超分处理功能
- 新增 ImageSRStage 类实现图像超分辨率处理 - 在 AioDeviceController 中启用图像超分和增强的 Stage - 修改 ImageEnhanceStage 配置检查逻辑,增加空值和占位符检测 - 为图像处理 Pipeline 添加超分 Stage - 增加 ImageSRStage 的单元测试覆盖各种配置和执行情况 - 实现百度云图像超分 API 的调用和结果处理逻辑
This commit is contained in:
@@ -9,6 +9,7 @@ import com.ycwl.basic.image.pipeline.core.PipelineBuilder;
|
||||
import com.ycwl.basic.image.pipeline.core.PhotoProcessContext;
|
||||
import com.ycwl.basic.image.pipeline.stages.DownloadStage;
|
||||
import com.ycwl.basic.image.pipeline.stages.ImageEnhanceStage;
|
||||
import com.ycwl.basic.image.pipeline.stages.ImageSRStage;
|
||||
import com.ycwl.basic.image.pipeline.stages.SourcePhotoUpdateStage;
|
||||
import com.ycwl.basic.image.pipeline.stages.CleanupStage;
|
||||
import com.ycwl.basic.mapper.AioDeviceMapper;
|
||||
@@ -155,10 +156,9 @@ public class AioDeviceController {
|
||||
photo.getGoodsId(), photo.getUrl(), photo.getScenicId()
|
||||
);
|
||||
|
||||
// 启用图像增强Stage
|
||||
Map<String, Boolean> stageConfig = new HashMap<>();
|
||||
stageConfig.put("image_enhance", true);
|
||||
context.loadStageConfig(null, stageConfig);
|
||||
// 启用图像增强和超分的Stage
|
||||
context.enableStage("image_enhance");
|
||||
context.enableStage("image_sr");
|
||||
|
||||
// 执行Pipeline
|
||||
boolean success = superResolutionPipeline.execute(context);
|
||||
@@ -243,7 +243,7 @@ public class AioDeviceController {
|
||||
|
||||
return new PipelineBuilder<PhotoProcessContext>("SourcePhotoSuperResolutionPipeline")
|
||||
.addStage(new DownloadStage()) // 1. 下载图片
|
||||
.addStage(new ImageEnhanceStage(config)) // 2. 图像增强(超分)
|
||||
.addStage(new ImageEnhanceStage(config)).addStage(new ImageSRStage(config)) // 2. 图像增强(超分)
|
||||
.addStage(new SourcePhotoUpdateStage(sourceService, sourceId)) // 3. 上传并更新数据库
|
||||
.addStage(new CleanupStage()) // 4. 清理临时文件
|
||||
.build();
|
||||
|
||||
Reference in New Issue
Block a user