|
|
|
|
@ -534,8 +534,6 @@ class ImageService:
|
|
|
|
|
def _get_evaluate_preview(task):
|
|
|
|
|
"""获取评估任务的预览图片"""
|
|
|
|
|
images = {
|
|
|
|
|
'original_generate': [],
|
|
|
|
|
'perturbed_generate': [],
|
|
|
|
|
'report': []
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
@ -547,17 +545,5 @@ class ImageService:
|
|
|
|
|
image_types_id=report_type.image_types_id
|
|
|
|
|
).all()
|
|
|
|
|
images['report'] = [ImageService.image_to_base64(img) for img in reports]
|
|
|
|
|
|
|
|
|
|
# 获取关联的微调任务生成图
|
|
|
|
|
if task.evaluation and task.evaluation.finetune_task_id:
|
|
|
|
|
finetune_task_id = task.evaluation.finetune_task_id
|
|
|
|
|
|
|
|
|
|
for type_code in ['original_generate', 'perturbed_generate']:
|
|
|
|
|
img_type = ImageType.query.filter_by(image_code=type_code).first()
|
|
|
|
|
if img_type:
|
|
|
|
|
generated = Image.query.filter_by(
|
|
|
|
|
task_id=finetune_task_id,
|
|
|
|
|
image_types_id=img_type.image_types_id
|
|
|
|
|
).all()
|
|
|
|
|
images[type_code] = [ImageService.image_to_base64(img) for img in generated]
|
|
|
|
|
return images
|