From 82016e32fc13423342cfa380931e4296a517ea4b Mon Sep 17 00:00:00 2001 From: ThuanNaN Date: Mon, 13 Jan 2025 11:12:55 +0700 Subject: [PATCH] update prompt ins --- minigpt4/datasets/datasets/mvtec_dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/minigpt4/datasets/datasets/mvtec_dataset.py b/minigpt4/datasets/datasets/mvtec_dataset.py index 893414a..705e774 100644 --- a/minigpt4/datasets/datasets/mvtec_dataset.py +++ b/minigpt4/datasets/datasets/mvtec_dataset.py @@ -35,11 +35,11 @@ class MVTecDataset(Dataset): image = Image.open(image_path).convert("RGB") image = self.vis_processor(image) - input = "detect a defect or not-defect object and return the bounding boxes and its label. If not, bound around the object." + input = "a defect or not-defect object and return the bounding boxes and its label. If not, bound around the object." 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)