You've already forked FrameTour-BE
- 新增 BceImageSR 类实现图像超分辨率增强 - 修改 BceImageEnhancer 类中的增强方法 -增加图像超分辨率增强的相关配置和接口
This commit is contained in:
@@ -33,7 +33,7 @@ public class BceImageEnhancer implements IEnhancer {
|
|||||||
public MultipartFile enhance(String url) {
|
public MultipartFile enhance(String url) {
|
||||||
AipImageProcess client = getClient();
|
AipImageProcess client = getClient();
|
||||||
HashMap<String, String> options = new HashMap<>();
|
HashMap<String, String> options = new HashMap<>();
|
||||||
JSONObject jsonObject = client.imageQualityEnhance(url, options);
|
JSONObject jsonObject = client.imageDefinitionEnhance(url, options);
|
||||||
return ImageUtils.base64ToMultipartFile(jsonObject.getString("image"));
|
return ImageUtils.base64ToMultipartFile(jsonObject.getString("image"));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@@ -0,0 +1,43 @@
|
|||||||
|
package com.ycwl.basic.image.enhancer.adapter;
|
||||||
|
|
||||||
|
import com.baidu.aip.imageprocess.AipImageProcess;
|
||||||
|
import com.ycwl.basic.image.enhancer.entity.BceEnhancerConfig;
|
||||||
|
import com.ycwl.basic.utils.ImageUtils;
|
||||||
|
import org.json.JSONObject;
|
||||||
|
import org.springframework.web.multipart.MultipartFile;
|
||||||
|
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.Map;
|
||||||
|
|
||||||
|
public class BceImageSR implements IEnhancer {
|
||||||
|
private BceEnhancerConfig config;
|
||||||
|
|
||||||
|
public boolean setConfig(BceEnhancerConfig config) {
|
||||||
|
this.config = config;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public boolean loadConfig(Map<String, String> _config) {
|
||||||
|
BceEnhancerConfig config = new BceEnhancerConfig();
|
||||||
|
config.setAppId(_config.get("appId"));
|
||||||
|
config.setApiKey(_config.get("apiKey"));
|
||||||
|
config.setSecretKey(_config.get("secretKey"));
|
||||||
|
config.setQps(Float.parseFloat(_config.get("qps")));
|
||||||
|
this.config = config;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public MultipartFile enhance(String url) {
|
||||||
|
AipImageProcess client = getClient();
|
||||||
|
HashMap<String, String> options = new HashMap<>();
|
||||||
|
JSONObject jsonObject = client.imageQualityEnhance(url, options);
|
||||||
|
return ImageUtils.base64ToMultipartFile(jsonObject.getString("image"));
|
||||||
|
}
|
||||||
|
|
||||||
|
public AipImageProcess getClient() {
|
||||||
|
AipImageProcess client = new AipImageProcess(config.getAppId(), config.getApiKey(), config.getSecretKey());
|
||||||
|
return client;
|
||||||
|
}
|
||||||
|
}
|
Reference in New Issue
Block a user