HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Simplify usage, remove YOLO.new() and YOLO.load()

Browse source
This commit is contained in:
Glenn Jocher 2023-01-02 15:42:01 +01:00
parent 7b21a87da3
commit d76d7af566
5 changed files with 33 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/check_flops.py
View file

@ -4,7 +4,7 @@ from ultralytics.yolo.utils import ROOT
if __name__ == "__main__":
for m in list((ROOT / 'yolo/v8/models').rglob('*.yaml')):
try:
YOLO.new(m.name, verbose=True)
YOLO(m.name, verbose=True)
except Exception as e:
print(f'ERROR for {m}: {e}')