Skip to content

Commit

Permalink
Update detect.py
Browse files Browse the repository at this point in the history
  • Loading branch information
sunnychenxiwang authored Feb 20, 2024
1 parent dc831c4 commit 8f7150e
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion pipeline/tool/detect.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down Expand Up @@ -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
return result

0 comments on commit 8f7150e

Please sign in to comment.