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 -------------------------------------------------------------------------------------------------------
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():

2
ultralytics/__init__.py
View File

@ -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

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

@ -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

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

@ -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)