add docker file, accelerate inference using cv2

2025-01-02 12:02:08 -08:00
parent d0c163cd02
commit 36b0cbea71
7 changed files with 320 additions and 26 deletions
--- a/omniparser.py
+++ b/omniparser.py
@@ -1,4 +1,4 @@
-from utils import get_som_labeled_img, check_ocr_box, get_caption_model_processor,  get_dino_model, get_yolo_model
+from utils import get_som_labeled_img, check_ocr_box, get_yolo_model
 import torch
 from ultralytics import YOLO
 from PIL import Image