RTDETRDetectionModel TorchScript, ONNX Predict and Val support (#8818)

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-03-09 23:25:01 +01:00 committed by GitHub
parent 911d18e4f3
commit af6c02c39b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 52 additions and 7 deletions

View file

@ -493,7 +493,7 @@ def check_file(file, suffix="", download=True, hard=True):
downloads.safe_download(url=url, file=file, unzip=False)
return file
else: # search
files = glob.glob(str(ROOT / "cfg" / "**" / file), recursive=True) # find file
files = glob.glob(str(ROOT / "**" / file), recursive=True) or glob.glob(str(ROOT.parent / file)) # find file
if not files and hard:
raise FileNotFoundError(f"'{file}' does not exist")
elif len(files) > 1 and hard: