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

Update metrics names (#85)

Browse source
This commit is contained in:
Glenn Jocher 2022-12-24 02:32:24 +01:00 • committed by GitHub
parent 6432afc5f9
commit 248d54ca03
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
9 changed files with 30 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/yolo/data/dataset.py
View file

@ -100,7 +100,7 @@ class YOLODataset(BaseDataset):
self.label_files = img2label_paths(self.im_files)
cache_path = Path(self.label_files[0]).parent.with_suffix(".cache")
try:
cache, exists = np.load(cache_path, allow_pickle=True).item(), True # load dict
cache, exists = np.load(str(cache_path), allow_pickle=True).item(), True # load dict
assert cache["version"] == self.cache_version # matches current version
assert cache["hash"] == get_hash(self.label_files + self.im_files) # identical hash
except Exception: