Unverified 提交 52d3a9ae authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

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

Fix detect.py --update Resolves https://github.com/ultralytics/yolov5/issues/8776
上级 e309a855
......@@ -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():
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论