Code Refactor ruff check --fix --extend-select I (#13093)

* Refactor code for speed and clarity

* Auto-format by https://ultralytics.com/actions

* Update train.py

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

* Update README.md

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

---------

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
Co-authored-by: Ultralytics Assistant <135830346+UltralyticsAssistant@users.noreply.github.com>

.github/workflows/merge-main-into-prs.yml

@@ -10,47 +10,47 @@ on:
     branches:
       - main
       - master
-  
+
 jobs:
   Merge:
     if: github.repository == 'ultralytics/yolov5'
     runs-on: ubuntu-latest
     steps:
-    - name: Checkout repository
+      - name: Checkout repository
-      uses: actions/checkout@v4
+        uses: actions/checkout@v4
-      with:
+        with:
-        fetch-depth: 0
+          fetch-depth: 0
-    - uses: actions/setup-python@v5
+      - uses: actions/setup-python@v5
-      with:
+        with:
-        python-version: "3.11"
+          python-version: "3.11"
-        cache: "pip" # caching pip dependencies
+          cache: "pip" # caching pip dependencies
-    - name: Install requirements
+      - name: Install requirements
-      run: |
+        run: |
-        pip install pygithub
+          pip install pygithub
-    - name: Merge main into PRs
+      - name: Merge main into PRs
-      shell: python
+        shell: python
-      run: |
+        run: |
-        from github import Github
+          from github import Github
-        import os
+          import os
-        
+
-        # Authenticate with the GitHub Token
+          # Authenticate with the GitHub Token
-        g = Github(os.getenv('GITHUB_TOKEN'))
+          g = Github(os.getenv('GITHUB_TOKEN'))
-        
+
-        # Get the repository dynamically
+          # Get the repository dynamically
-        repo = g.get_repo(os.getenv('GITHUB_REPOSITORY'))
+          repo = g.get_repo(os.getenv('GITHUB_REPOSITORY'))
-        
+
-        # List all open pull requests
+          # List all open pull requests
-        open_pulls = repo.get_pulls(state='open', sort='created')
+          open_pulls = repo.get_pulls(state='open', sort='created')
-        
+
-        for pr in open_pulls:
+          for pr in open_pulls:
-            # Compare PR head with main to see if it's behind
+              # Compare PR head with main to see if it's behind
-            try:
+              try:
-                # Merge main into the PR branch
+                  # Merge main into the PR branch
-                success = pr.update_branch()
+                  success = pr.update_branch()
-                assert success, "Branch update failed"
+                  assert success, "Branch update failed"
-                print(f"Merged 'master' into PR #{pr.number} ({pr.head.ref}) successfully.")
+                  print(f"Merged 'master' into PR #{pr.number} ({pr.head.ref}) successfully.")
-            except Exception as e:
+              except Exception as e:
-                print(f"Could not merge 'master' into PR #{pr.number} ({pr.head.ref}): {e}")
+                  print(f"Could not merge 'master' into PR #{pr.number} ({pr.head.ref}): {e}")
-      env:
+        env:
-        GITHUB_TOKEN: ${{ secrets.PERSONAL_ACCESS_TOKEN }}
+          GITHUB_TOKEN: ${{ secrets.PERSONAL_ACCESS_TOKEN }}
-        GITHUB_REPOSITORY: ${{ github.repository }}
+          GITHUB_REPOSITORY: ${{ github.repository }}

classify/train.py

@@ -178,7 +178,9 @@ def train(opt, device):
     # Scheduler
     lrf = 0.01  # final lr (fraction of lr0)
     # lf = lambda x: ((1 + math.cos(x * math.pi / epochs)) / 2) * (1 - lrf) + lrf  # cosine
-    lf = lambda x: (1 - x / epochs) * (1 - lrf) + lrf  # linear
+    def lf(x):
+        return (1 - x / epochs) * (1 - lrf) + lrf  # linear
+
     scheduler = lr_scheduler.LambdaLR(optimizer, lr_lambda=lf)
     # scheduler = lr_scheduler.OneCycleLR(optimizer, max_lr=lr0, total_steps=epochs, pct_start=0.1,
     #                                    final_div_factor=1 / 25 / lrf)

models/yolo.py

@@ -244,7 +244,10 @@ class DetectionModel(BaseModel):
         if isinstance(m, (Detect, Segment)):
             s = 256  # 2x min stride
             m.inplace = self.inplace
-            forward = lambda x: self.forward(x)[0] if isinstance(m, Segment) else self.forward(x)
+
+            def forward(x):
+                return self.forward(x)[0] if isinstance(m, Segment) else self.forward(x)
+
             m.stride = torch.tensor([s / x.shape[-2] for x in forward(torch.zeros(1, ch, s, s))])  # forward
             check_anchor_order(m)
             m.anchors /= m.stride.view(-1, 1, 1)

segment/train.py

@@ -214,7 +214,10 @@ def train(hyp, opt, device, callbacks):
     if opt.cos_lr:
         lf = one_cycle(1, hyp["lrf"], epochs)  # cosine 1->hyp['lrf']
     else:
-        lf = lambda x: (1 - x / epochs) * (1.0 - hyp["lrf"]) + hyp["lrf"]  # linear
+
+        def lf(x):
+            return (1 - x / epochs) * (1.0 - hyp["lrf"]) + hyp["lrf"]  # linear
+
     scheduler = lr_scheduler.LambdaLR(optimizer, lr_lambda=lf)  # plot_lr_scheduler(optimizer, scheduler, epochs)
 
     # EMA

train.py

@@ -224,7 +224,10 @@ def train(hyp, opt, device, callbacks):
     if opt.cos_lr:
         lf = one_cycle(1, hyp["lrf"], epochs)  # cosine 1->hyp['lrf']
     else:
-        lf = lambda x: (1 - x / epochs) * (1.0 - hyp["lrf"]) + hyp["lrf"]  # linear
+
+        def lf(x):
+            return (1 - x / epochs) * (1.0 - hyp["lrf"]) + hyp["lrf"]  # linear
+
     scheduler = lr_scheduler.LambdaLR(optimizer, lr_lambda=lf)  # plot_lr_scheduler(optimizer, scheduler, epochs)
 
     # EMA

utils/loggers/__init__.py

@@ -21,7 +21,10 @@ RANK = int(os.getenv("RANK", -1))
 try:
     from torch.utils.tensorboard import SummaryWriter
 except ImportError:
-    SummaryWriter = lambda *args: None  # None = SummaryWriter(str)
+
+    def SummaryWriter(*args):
+        return None  # None = SummaryWriter(str)
+
 
 try:
     import wandb

utils/loggers/clearml/README.md

@@ -16,16 +16,10 @@
 
 🔭 Turn your newly trained <b>YOLOv5 model into an API</b> with just a few commands using ClearML Serving
 
-<br />
 And so much more. It's up to you how many of these tools you want to use, you can stick to the experiment manager, or chain them all together into an impressive pipeline!
-<br />
-<br />
 
 ![ClearML scalars dashboard](https://github.com/thepycoder/clearml_screenshots/raw/main/experiment_manager_with_compare.gif)
 
-<br />
-<br />
-
 ## 🦾 Setting Things Up
 
 To keep track of your experiments and/or data, ClearML needs to communicate to a server. You have 2 options to get one:
@@ -46,8 +40,6 @@ Either sign up for free to the [ClearML Hosted Service](https://cutt.ly/yolov5-t
 
 That's it! You're done 😎
 
-<br />
-
 ## 🚀 Training YOLOv5 With ClearML
 
 To enable ClearML experiment tracking, simply install the ClearML pip package.
@@ -89,8 +81,6 @@ That's a lot right? 🤯 Now, we can visualize all of this information in the Cl
 
 There even more we can do with all of this information, like hyperparameter optimization and remote execution, so keep reading if you want to see how that works!
 
-<br />
-
 ## 🔗 Dataset Version Management
 
 Versioning your data separately from your code is generally a good idea and makes it easy to acquire the latest version too. This repository supports supplying a dataset version ID, and it will make sure to get the data if it's not there yet. Next to that, this workflow also saves the used dataset ID as part of the task parameters, so you will always know for sure which data was used in which experiment!
@@ -157,8 +147,6 @@ Now that you have a ClearML dataset, you can very simply use it to train custom
 python train.py --img 640 --batch 16 --epochs 3 --data clearml://<your_dataset_id> --weights yolov5s.pt --cache
 ```
 
-<br />
-
 ## 👀 Hyperparameter Optimization
 
 Now that we have our experiments and data versioned, it's time to take a look at what we can build on top!