From 3718f9ff6981176481f53fa172fe3f680cfbef4f Mon Sep 17 00:00:00 2001 From: ThuanNaN Date: Mon, 13 Jan 2025 06:48:23 +0700 Subject: [PATCH] update anwser ins --- minigpt4/datasets/datasets/mvtec_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/minigpt4/datasets/datasets/mvtec_dataset.py b/minigpt4/datasets/datasets/mvtec_dataset.py index 287f09b..893414a 100644 --- a/minigpt4/datasets/datasets/mvtec_dataset.py +++ b/minigpt4/datasets/datasets/mvtec_dataset.py @@ -39,7 +39,7 @@ class MVTecDataset(Dataset): ans_defect = "defect" if info["is_broken"] == True else "non-defect" ans_para = f"

{ans_cls}-{ans_defect}

" - answer = f"{ans_para}<{gt_bbox[0]}> <{gt_bbox[1]}><{gt_bbox[2]}><{gt_bbox[3]}>" + answer = f"{ans_para} {{<{gt_bbox[0]}><{gt_bbox[1]}><{gt_bbox[2]}><{gt_bbox[3]}>}}" instruction = random.choice(self.instruction_pool).format(input) instruction = " {} ".format(instruction)