8.0.15 CLI float bug fix (#567)

single_channel
Glenn Jocher 2 years ago committed by GitHub
parent 21b701c4ea
commit 3c4de102f6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -48,7 +48,7 @@ def test_val_classify():
# Predict checks -------------------------------------------------------------------------------------------------------
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():

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

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

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

Loading…
Cancel
Save