|
@@ -4,6 +4,10 @@ const { Controller } = require('ee-core');
|
|
|
const Log = require('ee-core/log');
|
|
const Log = require('ee-core/log');
|
|
|
const { dialog } = require('electron');
|
|
const { dialog } = require('electron');
|
|
|
const { checkSelectImages,segmentImages,modelFormSegment } = require('../api/imageMatting')
|
|
const { checkSelectImages,segmentImages,modelFormSegment } = require('../api/imageMatting')
|
|
|
|
|
+const errData = {
|
|
|
|
|
+ msg :'请求失败,请联系管理员',
|
|
|
|
|
+ code:999
|
|
|
|
|
+}
|
|
|
|
|
|
|
|
/**
|
|
/**
|
|
|
* example
|
|
* example
|
|
@@ -67,7 +71,10 @@ class ImageMattingController extends Controller {
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- const res = await checkSelectImages(selectParams)
|
|
|
|
|
|
|
+ const res = await checkSelectImages(selectParams).catch(e=>{
|
|
|
|
|
+ Log.info(e);
|
|
|
|
|
+ return errData
|
|
|
|
|
+ })
|
|
|
|
|
|
|
|
|
|
|
|
|
Log.info(res.data);
|
|
Log.info(res.data);
|
|
@@ -93,7 +100,10 @@ class ImageMattingController extends Controller {
|
|
|
delete modelParams.out_width
|
|
delete modelParams.out_width
|
|
|
delete modelParams.out_height
|
|
delete modelParams.out_height
|
|
|
}
|
|
}
|
|
|
- const img = await modelFormSegment(modelParams)
|
|
|
|
|
|
|
+ const img = await modelFormSegment(modelParams).catch(e=>{
|
|
|
|
|
+ Log.info(e);
|
|
|
|
|
+ return errData
|
|
|
|
|
+ })
|
|
|
|
|
|
|
|
if(img.data){
|
|
if(img.data){
|
|
|
return img.data
|
|
return img.data
|
|
@@ -109,6 +119,9 @@ class ImageMattingController extends Controller {
|
|
|
"segment_type": params.segment_type, //抠图精细度;0普通;1精细
|
|
"segment_type": params.segment_type, //抠图精细度;0普通;1精细
|
|
|
"output_type": params.output_type, //出图模式 ;0透明;1白底
|
|
"output_type": params.output_type, //出图模式 ;0透明;1白底
|
|
|
"need_cutout_images": res.data.data
|
|
"need_cutout_images": res.data.data
|
|
|
|
|
+ }).catch(e=>{
|
|
|
|
|
+ Log.info(e);
|
|
|
|
|
+ return errData
|
|
|
})
|
|
})
|
|
|
if(img.data){
|
|
if(img.data){
|
|
|
return img.data
|
|
return img.data
|