Register
Login
Resources
Docs Blog Datasets Glossary Case Studies Tutorials & Webinars
Product
Data Engine LLMs Platform Enterprise
Pricing Explore
Connect to our Discord channel

#20413 YOLOE: Fix visual prompt training

Merged
Ghost merged 1 commits into Ultralytics:main from ultralytics:yoloe-vp-fix
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
  1. # Ultralytics ๐Ÿš€ AGPL-3.0 License - https://ultralytics.com/license
  2. __version__ = "8.3.119"
  3. import os
  4. # Set ENV variables (place before imports)
  5. if not os.environ.get("OMP_NUM_THREADS"):
  6. os.environ["OMP_NUM_THREADS"] = "1" # default for reduced CPU utilization during training
  7. from ultralytics.models import NAS, RTDETR, SAM, YOLO, YOLOE, FastSAM, YOLOWorld
  8. from ultralytics.utils import ASSETS, SETTINGS
  9. from ultralytics.utils.checks import check_yolo as checks
  10. from ultralytics.utils.downloads import download
  11. settings = SETTINGS
  12. __all__ = (
  13. "__version__",
  14. "ASSETS",
  15. "YOLO",
  16. "YOLOWorld",
  17. "YOLOE",
  18. "NAS",
  19. "SAM",
  20. "FastSAM",
  21. "RTDETR",
  22. "checks",
  23. "download",
  24. "settings",
  25. )
Discard
Tip!

Press p or to see the previous file or, n or to see the next file