|
@@ -66,8 +66,12 @@ def label_verification(model_filename: str, framework: str='pytorch', mode: str=
|
|
|
code=-2,
|
|
|
message=f"不支持的组合: framework={framework}, mode={mode}, model_type={model_type}"
|
|
|
)
|
|
|
+ if mode == 'whitebox' and model_type == 'vggnet':
|
|
|
+ result = processor_class.process((3, 7))
|
|
|
+ else:
|
|
|
+ result = processor_class.process()
|
|
|
|
|
|
- result = processor_class.process()
|
|
|
+ return result
|
|
|
|
|
|
except Exception as e:
|
|
|
raise BusinessException(code=-1, message=str(e))
|