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 = "