Extra preprocessed input (Fixed issues) ultralytics#20

manole-alexandru · Apr 25, 2023 · 37a5080 · 37a5080
1 parent 3384090
commit 37a5080
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/models/yolov5m.yaml b/models/yolov5m.yaml
@@ -2,6 +2,7 @@
 
 # Parameters
 nc: 80  # number of classes
+ch: 4
 depth_multiple: 0.67  # model depth multiple
 width_multiple: 0.75  # layer channel multiple
 anchors:

diff --git a/train.py b/train.py
@@ -553,7 +553,7 @@ def parse_opt(known=False):
 
 
 def main(opt, callbacks=Callbacks()):
-    print('\n---------- VERSION:', '#0019', '----------\n')
+    print('\n---------- VERSION:', '#0020', '----------\n')
     # Checks
     if RANK in {-1, 0}:
         print_args(vars(opt))