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)