ultralytics 8.1.38__py3-none-any.whl → 8.1.40__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of ultralytics might be problematic. Click here for more details.

Files changed (58) hide show
  1. ultralytics/__init__.py +1 -1
  2. ultralytics/cfg/__init__.py +3 -3
  3. ultralytics/cfg/datasets/lvis.yaml +1239 -0
  4. ultralytics/data/__init__.py +18 -2
  5. ultralytics/data/augment.py +124 -3
  6. ultralytics/data/base.py +2 -2
  7. ultralytics/data/build.py +25 -3
  8. ultralytics/data/converter.py +24 -6
  9. ultralytics/data/dataset.py +142 -27
  10. ultralytics/data/loaders.py +11 -8
  11. ultralytics/data/split_dota.py +1 -1
  12. ultralytics/data/utils.py +33 -8
  13. ultralytics/engine/exporter.py +3 -3
  14. ultralytics/engine/model.py +6 -3
  15. ultralytics/engine/results.py +2 -2
  16. ultralytics/engine/trainer.py +59 -55
  17. ultralytics/engine/validator.py +2 -2
  18. ultralytics/hub/utils.py +1 -1
  19. ultralytics/models/fastsam/model.py +1 -1
  20. ultralytics/models/fastsam/prompt.py +4 -5
  21. ultralytics/models/nas/model.py +1 -1
  22. ultralytics/models/sam/model.py +1 -1
  23. ultralytics/models/sam/modules/tiny_encoder.py +1 -1
  24. ultralytics/models/yolo/__init__.py +2 -2
  25. ultralytics/models/yolo/classify/train.py +1 -1
  26. ultralytics/models/yolo/detect/train.py +1 -1
  27. ultralytics/models/yolo/detect/val.py +36 -17
  28. ultralytics/models/yolo/model.py +1 -0
  29. ultralytics/models/yolo/world/__init__.py +5 -0
  30. ultralytics/models/yolo/world/train.py +92 -0
  31. ultralytics/models/yolo/world/train_world.py +108 -0
  32. ultralytics/nn/autobackend.py +5 -5
  33. ultralytics/nn/modules/block.py +4 -2
  34. ultralytics/nn/modules/conv.py +1 -1
  35. ultralytics/nn/modules/head.py +13 -4
  36. ultralytics/nn/tasks.py +30 -14
  37. ultralytics/solutions/ai_gym.py +1 -1
  38. ultralytics/solutions/heatmap.py +85 -47
  39. ultralytics/solutions/object_counter.py +79 -64
  40. ultralytics/trackers/byte_tracker.py +1 -1
  41. ultralytics/trackers/track.py +1 -1
  42. ultralytics/trackers/utils/gmc.py +1 -1
  43. ultralytics/utils/__init__.py +4 -4
  44. ultralytics/utils/benchmarks.py +2 -2
  45. ultralytics/utils/callbacks/comet.py +1 -1
  46. ultralytics/utils/callbacks/mlflow.py +1 -1
  47. ultralytics/utils/checks.py +3 -3
  48. ultralytics/utils/downloads.py +2 -2
  49. ultralytics/utils/loss.py +1 -1
  50. ultralytics/utils/metrics.py +1 -1
  51. ultralytics/utils/plotting.py +36 -22
  52. ultralytics/utils/torch_utils.py +17 -3
  53. {ultralytics-8.1.38.dist-info → ultralytics-8.1.40.dist-info}/METADATA +1 -1
  54. {ultralytics-8.1.38.dist-info → ultralytics-8.1.40.dist-info}/RECORD +58 -54
  55. {ultralytics-8.1.38.dist-info → ultralytics-8.1.40.dist-info}/LICENSE +0 -0
  56. {ultralytics-8.1.38.dist-info → ultralytics-8.1.40.dist-info}/WHEEL +0 -0
  57. {ultralytics-8.1.38.dist-info → ultralytics-8.1.40.dist-info}/entry_points.txt +0 -0
  58. {ultralytics-8.1.38.dist-info → ultralytics-8.1.40.dist-info}/top_level.txt +0 -0
ultralytics/__init__.py CHANGED
@@ -1,6 +1,6 @@
1
1
  # Ultralytics YOLO 🚀, AGPL-3.0 license
2
2
 
3
- __version__ = "8.1.38"
3
+ __version__ = "8.1.40"
4
4
 
5
5
  from ultralytics.data.explorer.explorer import Explorer
6
6
  from ultralytics.models import RTDETR, SAM, YOLO, YOLOWorld
ultralytics/cfg/__init__.py CHANGED
@@ -272,7 +272,7 @@ def get_save_dir(args, name=None):
272
272
 
273
273
  project = args.project or (ROOT.parent / "tests/tmp/runs" if TESTS_RUNNING else RUNS_DIR) / args.task
274
274
  name = name or args.name or f"{args.mode}"
275
- save_dir = increment_path(Path(project) / name, exist_ok=args.exist_ok if RANK in (-1, 0) else True)
275
+ save_dir = increment_path(Path(project) / name, exist_ok=args.exist_ok if RANK in {-1, 0} else True)
276
276
 
277
277
  return Path(save_dir)
278
278
 
@@ -566,10 +566,10 @@ def entrypoint(debug=""):
566
566
  task = model.task
567
567
 
568
568
  # Mode
569
- if mode in ("predict", "track") and "source" not in overrides:
569
+ if mode in {"predict", "track"} and "source" not in overrides:
570
570
  overrides["source"] = DEFAULT_CFG.source or ASSETS
571
571
  LOGGER.warning(f"WARNING ⚠️ 'source' argument is missing. Using default 'source={overrides['source']}'.")
572
- elif mode in ("train", "val"):
572
+ elif mode in {"train", "val"}:
573
573
  if "data" not in overrides and "resume" not in overrides:
574
574
  overrides["data"] = DEFAULT_CFG.data or TASK2DATA.get(task or DEFAULT_CFG.task, DEFAULT_CFG.data)
575
575
  LOGGER.warning(f"WARNING ⚠️ 'data' argument is missing. Using default 'data={overrides['data']}'.")