田运杰 10 mesi fa
parent
commit
5ebe774c41
1 ha cambiato i file con 2 aggiunte e 2 eliminazioni
  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
                 if scale in "mlx":
                     args[3] = True
-            if m is A2C2f:  # for M/L/X sizes
+            if m is A2C2f: 
                 legacy = False
-                if scale in "lx":
+                if scale in "lx":  # for L/X sizes
                     args.append(True)
                     args.append(1.2)
         elif m is AIFI: