YOLOv5 v5.0 Release patch 1 (#2764)

* torch.jit.trace(model, img, strict=False)

* Update check_file()

* Update hubconf.py

* Update README.md
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-04-11 23:11:43 +02:00
1 parent f5b8f7d54c
commit 0f395b3e3b
4 files changed
+8 -8

No files matched your search

+3 -3
View File
@@ -144,12 +144,12 @@ def check_imshow():
def check_file(file):
# Search for file if not found
if os.path.isfile(file) or file == '':
if Path(file).is_file() or file == '':
return file
else:
files = glob.glob('./**/' + file, recursive=True) # find file
assert len(files), 'File Not Found: %s' % file # assert file was found
assert len(files) == 1, "Multiple files match '%s', specify exact path: %s" % (file, files) # assert unique
assert len(files), f'File Not Found: {file}' # assert file was found
assert len(files) == 1, f"Multiple files match '{file}', specify exact path: {files}" # assert unique
return files[0] # return file