diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml index 1b4e3c7..ee4b192 100644 --- a/.github/workflows/ci.yaml +++ b/.github/workflows/ci.yaml @@ -5,9 +5,9 @@ name: Ultralytics CI on: push: - branches: [ main ] + branches: [main] pull_request: - branches: [ main ] + branches: [main] # schedule: # - cron: '0 0 * * *' # runs at 00:00 UTC every day @@ -18,24 +18,18 @@ jobs: strategy: fail-fast: false matrix: - os: [ ubuntu-latest ] - python-version: [ '3.10' ] - model: [ yolov8n ] - torch: [ latest ] -# include: -# - os: ubuntu-latest -# python-version: '3.7' # '3.6.8' min -# model: yolov8n -# - os: ubuntu-latest -# python-version: '3.8' -# model: yolov8n -# - os: ubuntu-latest -# python-version: '3.9' -# model: yolov8n -# - os: ubuntu-latest -# python-version: '3.8' # torch 1.7.0 requires python >=3.6, <=3.8 -# model: yolov8n -# torch: '1.7.0' # min torch version CI https://pypi.org/project/torchvision/ + os: [ubuntu-latest, windows-latest, macos-latest] + python-version: ['3.10'] + model: [yolov8n] + torch: [latest] + include: + - os: ubuntu-latest + python-version: '3.7' # '3.6.8' min + model: yolov8n + - os: ubuntu-latest + python-version: '3.8' # torch 1.7.0 requires python >=3.6, <=3.8 + model: yolov8n + torch: '1.7.0' # min torch version CI https://pypi.org/project/torchvision/ steps: - uses: actions/checkout@v3 - uses: actions/setup-python@v4 @@ -92,13 +86,16 @@ jobs: run: | yolo task=detect mode=train model=yolov8n.yaml data=coco128.yaml epochs=1 imgsz=64 yolo task=detect mode=val model=runs/detect/train/weights/last.pt imgsz=64 + yolo task=detect mode=predict model=runs/detect/train/weights/last.pt imgsz=64 source=ultralytics/assets/bus.jpg - name: Test segmentation shell: bash # for Windows compatibility run: | yolo task=segment mode=train model=yolov8n-seg.yaml data=coco128-seg.yaml epochs=1 imgsz=64 yolo task=segment mode=val model=runs/segment/train/weights/last.pt data=coco128-seg.yaml imgsz=64 + yolo task=segment mode=predict model=runs/segment/train/weights/last.pt imgsz=64 source=ultralytics/assets/bus.jpg - name: Test classification shell: bash # for Windows compatibility run: | yolo task=classify mode=train model=yolov8n-cls.yaml data=mnist160 epochs=1 imgsz=32 - yolo task=classify mode=val model=runs/classify/train/weights/last.pt data=mnist160 + yolo task=classify mode=val model=runs/classify/train/weights/last.pt data=mnist160 imgsz=32 + yolo task=classify mode=predict model=runs/classify/train/weights/last.pt imgsz=32 source=ultralytics/assets/bus.jpg diff --git a/.github/workflows/docker.yaml b/.github/workflows/docker.yaml index bcade98..8b816bb 100644 --- a/.github/workflows/docker.yaml +++ b/.github/workflows/docker.yaml @@ -5,7 +5,7 @@ name: Publish Docker Images on: push: - branches: [ none ] # TODO: replace with main + branches: [none] # TODO: replace with main jobs: docker: diff --git a/ultralytics/hub/utils.py b/ultralytics/hub/utils.py index 753afc7..1b92539 100644 --- a/ultralytics/hub/utils.py +++ b/ultralytics/hub/utils.py @@ -131,7 +131,7 @@ def smart_request(*args, retry=3, timeout=30, thread=True, code=-1, method="post @TryExcept() -def sync_analytics(cfg, all_keys=False, enabled=True): +def sync_analytics(cfg, all_keys=False, enabled=False): """ Sync analytics data if enabled in the global settings diff --git a/ultralytics/yolo/v8/classify/train.py b/ultralytics/yolo/v8/classify/train.py index bb03ba1..46660e1 100644 --- a/ultralytics/yolo/v8/classify/train.py +++ b/ultralytics/yolo/v8/classify/train.py @@ -72,7 +72,8 @@ class ClassificationTrainer(BaseTrainer): imgsz=self.args.imgsz, batch_size=batch_size if mode == "train" else (batch_size * 2), augment=mode == "train", - rank=rank) + rank=rank, + workers=self.args.workers) def preprocess_batch(self, batch): batch["img"] = batch["img"].to(self.device) diff --git a/ultralytics/yolo/v8/classify/val.py b/ultralytics/yolo/v8/classify/val.py index 9ccd710..a9895d4 100644 --- a/ultralytics/yolo/v8/classify/val.py +++ b/ultralytics/yolo/v8/classify/val.py @@ -36,7 +36,10 @@ class ClassificationValidator(BaseValidator): return self.metrics.results_dict def get_dataloader(self, dataset_path, batch_size): - return build_classification_dataloader(path=dataset_path, imgsz=self.args.imgsz, batch_size=batch_size) + return build_classification_dataloader(path=dataset_path, + imgsz=self.args.imgsz, + batch_size=batch_size, + workers=self.args.workers) def print_results(self): pf = '%22s' + '%11.3g' * len(self.metrics.keys) # print format