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

version 1.5

Browse Source
This commit is contained in:
yadonglu
2024-11-26 13:04:51 -08:00
parent 7021ad8917
commit 856b539e54
3 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
demo.ipynb
View File

@@ -20,9 +20,6 @@
"from PIL import Image\n",
"device = 'cuda'\n",
"\n",
"# som_model = get_yolo_model(model_path='weights/icon_detect/best.pt')\n",
"# som_model = get_yolo_model(model_path='/home/yadonglu/sandbox/data/yolo/runs/detect/yolo11l_som_detection_seq_10ep_b32_filter5more4/weights/best.pt')\n",
"# som_model = get_yolo_model('/home/yadonglu/sandbox/data/yolo/runs/detect/yolo11l_som_detection_seq_10ep_b24_filter5more1280/weights/best.pt')\n",
"som_model = get_yolo_model(model_path='weights/icon_detect_v1_5/best.pt')\n",
"\n",
"som_model.to(device)\n",