mirror of
https://github.com/Vision-CAIR/MiniGPT-4.git
synced 2025-04-05 02:20:47 +00:00
update prompt ins
This commit is contained in:
parent
8415c675a5
commit
82016e32fc
@ -35,7 +35,7 @@ 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>"
|
||||
|
Loading…
Reference in New Issue
Block a user