Dom/OmniParser
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix import path issues

Browse Source
This commit is contained in:
Thomas Dhome-Casanova
2025-01-30 06:18:24 +00:00
parent 124d9f6fb6
commit 41464ccf1c
3 changed files with 19 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
demo.ipynb
View File

@@ -14,7 +14,7 @@
}
],
"source": [
"from utils import get_som_labeled_img, check_ocr_box, get_caption_model_processor, get_yolo_model\n",
"from util.utils import get_som_labeled_img, check_ocr_box, get_caption_model_processor, get_yolo_model\n",
"import torch\n",
"from ultralytics import YOLO\n",
"from PIL import Image\n",
@@ -48,9 +48,9 @@
"source": [
"# two choices for caption model: fine-tuned blip2 or florence2\n",
"import importlib\n",
"import utils\n",
"importlib.reload(utils)\n",
"from utils import get_som_labeled_img, check_ocr_box, get_caption_model_processor, get_yolo_model\n",
"import util.utils\n",
"importlib.reload(util.utils)\n",
"from util.utils import get_som_labeled_img, check_ocr_box, get_caption_model_processor, get_yolo_model\n",
"# caption_model_processor = get_caption_model_processor(model_name=\"blip2\", model_name_or_path=\"weights/icon_caption_blip2\", device=device)\n",
"caption_model_processor = get_caption_model_processor(model_name=\"florence2\", model_name_or_path=\"weights/icon_caption_florence\", device=device)\n",
"\n"
@@ -102,9 +102,9 @@
"source": [
"# reload utils\n",
"import importlib\n",
"import utils\n",
"importlib.reload(utils)\n",
"from utils import get_som_labeled_img, check_ocr_box, get_caption_model_processor, get_yolo_model\n",
"import util.utils\n",
"importlib.reload(util.utils)\n",
"from util.utils import get_som_labeled_img, check_ocr_box, get_caption_model_processor, get_yolo_model\n",
"\n",
"image_path = 'imgs/google_page.png'\n",
"image_path = 'imgs/windows_home.png'\n",
@@ -167,9 +167,9 @@
"# run on cpu!!!\n",
"# reload utils\n",
"import importlib\n",
"import utils\n",
"importlib.reload(utils)\n",
"from utils import get_som_labeled_img, check_ocr_box, get_caption_model_processor, get_yolo_model\n",
"import util.utils\n",
"importlib.reload(util.utils)\n",
"from util.utils import get_som_labeled_img, check_ocr_box, get_caption_model_processor, get_yolo_model\n",
"\n",
"image_path = 'imgs/google_page.png'\n",
"image_path = 'imgs/windows_home.png'\n",
@@ -447,13 +447,6 @@
"source": [
"parsed_content_list[-1]"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {