chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-04 18:10:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

update prompt ins

Browse Source
This commit is contained in:
ThuanNaN 2025-01-13 11:12:55 +07:00
parent 8415c675a5
commit 82016e32fc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
minigpt4/datasets/datasets/mvtec_dataset.py
View File

@ -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"<p>{ans_cls}-{ans_defect}</p>"
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 = "<Img><ImageHere></Img> {} ".format(instruction)