xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 13:34:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

8.0.15 CLI float bug fix (#567)

Browse Source
This commit is contained in:
Glenn Jocher 2023-01-22 17:48:31 +01:00 committed by GitHub
parent 21b701c4ea
commit 3c4de102f6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_cli.py
View File

@ -48,7 +48,7 @@ def test_val_classify():
# Predict checks ------------------------------------------------------------------------------------------------------- # Predict checks -------------------------------------------------------------------------------------------------------
def test_predict_detect(): def test_predict_detect():
run(f"yolo predict detect model={MODEL}.pt source={ROOT / 'assets'}") run(f"yolo predict detect model={MODEL}.pt source={ROOT / 'assets'} imgsz=320 conf=0.25")
def test_predict_segment(): def test_predict_segment():

2
ultralytics/__init__.py
View File

@ -1,6 +1,6 @@
# Ultralytics YOLO 🚀, GPL-3.0 license # Ultralytics YOLO 🚀, GPL-3.0 license
__version__ = "8.0.14" __version__ = "8.0.15"
from ultralytics.yolo.engine.model import YOLO from ultralytics.yolo.engine.model import YOLO
from ultralytics.yolo.utils import ops from ultralytics.yolo.utils import ops

10
ultralytics/yolo/cfg/__init__.py
View File

@ -1,4 +1,5 @@
# Ultralytics YOLO 🚀, GPL-3.0 license # Ultralytics YOLO 🚀, GPL-3.0 license
import contextlib
import re import re
import shutil import shutil
import sys import sys
@ -154,16 +155,15 @@ def entrypoint(debug=False):
LOGGER.info(f"{PREFIX}Overriding {DEFAULT_CFG_PATH} with {v}") LOGGER.info(f"{PREFIX}Overriding {DEFAULT_CFG_PATH} with {v}")
overrides = {k: val for k, val in yaml_load(v).items() if k != 'cfg'} overrides = {k: val for k, val in yaml_load(v).items() if k != 'cfg'}
else: else:
if v.isnumeric(): if v.lower() == 'none':
v = eval(v)
elif v.lower() == 'none':
v = None v = None
elif v.lower() == 'true': elif v.lower() == 'true':
v = True v = True
elif v.lower() == 'false': elif v.lower() == 'false':
v = False v = False
elif ',' in v: else:
v = eval(v) with contextlib.suppress(Exception):
v = eval(v)
overrides[k] = v overrides[k] = v
except (NameError, SyntaxError, ValueError) as e: except (NameError, SyntaxError, ValueError) as e:
raise argument_error(a) from e raise argument_error(a) from e

1
ultralytics/yolo/engine/model.py
View File

@ -172,7 +172,6 @@ class YOLO:
args = get_cfg(cfg=DEFAULT_CFG, overrides=overrides) args = get_cfg(cfg=DEFAULT_CFG, overrides=overrides)
args.task = self.task args.task = self.task
print(args)
exporter = Exporter(overrides=args) exporter = Exporter(overrides=args)
exporter(model=self.model) exporter(model=self.model)