Fix detect.py --update list bug (ultralytics#8783)

Fix detect.py --update Resolves ultralytics#8776
libaos · Jul 29, 2022 · 52d3a9a · 52d3a9a
1 parent e309a85
commit 52d3a9a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/detect.py b/detect.py
@@ -210,7 +210,7 @@ def run(
         s = f"\n{len(list(save_dir.glob('labels/*.txt')))} labels saved to {save_dir / 'labels'}" if save_txt else ''
         LOGGER.info(f"Results saved to {colorstr('bold', save_dir)}{s}")
     if update:
-        strip_optimizer(weights)  # update model (to fix SourceChangeWarning)
+        strip_optimizer(weights[0])  # update model (to fix SourceChangeWarning)
 
 
 def parse_opt():