From 8f7150e2e3932281555cb233c4adaaa66fd18b09 Mon Sep 17 00:00:00 2001
From: sunnychenxiwang <153065748+sunnychenxiwang@users.noreply.github.com>
Date: Tue, 20 Feb 2024 17:16:41 +0800
Subject: [PATCH] Update detect.py

---
 pipeline/tool/detect.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/pipeline/tool/detect.py b/pipeline/tool/detect.py
index b7fda23..3d645da 100644
--- a/pipeline/tool/detect.py
+++ b/pipeline/tool/detect.py
@@ -1,3 +1,8 @@
+# Utilizing the following code snippet based on the algorithm proposed in the paper:
+# Title: "Woodpecker: Hallucination correction for multimodal large language models"
+# Author: Yin et al.
+# Original code source: https://github.com/BradyFU/Woodpecker/blob/main/models/detector.py
+
 import yaml
 import torch
 import os
@@ -40,4 +45,4 @@ def execute(self, image_path, content, box_threshold, text_threshold,save_path):
                 image_path_list.append(crop_path)
             result["save_path"] = image_path_list
         
-        return result
\ No newline at end of file
+        return result