田运杰 11 ay önce
ebeveyn
işleme
5ebe774c41
1 değiştirilmiş dosya ile 2 ekleme ve 2 silme
  1. 2 2
      ultralytics/nn/tasks.py

+ 2 - 2
ultralytics/nn/tasks.py

@@ -1031,9 +1031,9 @@ def parse_model(d, ch, verbose=True):  # model_dict, input_channels(3)
                 legacy = False
                 legacy = False
                 if scale in "mlx":
                 if scale in "mlx":
                     args[3] = True
                     args[3] = True
-            if m is A2C2f:  # for M/L/X sizes
+            if m is A2C2f: 
                 legacy = False
                 legacy = False
-                if scale in "lx":
+                if scale in "lx":  # for L/X sizes
                     args.append(True)
                     args.append(True)
                     args.append(1.2)
                     args.append(1.2)
         elif m is AIFI:
         elif m is AIFI: