Browse Source

Update cache comments (#8414)

* Update cache comments

For better readability

* Update dataloaders.py
modifyDataloader
Glenn Jocher GitHub 2 years ago
parent
commit
898332433a
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      utils/dataloaders.py

+ 3
- 3
utils/dataloaders.py View File

@@ -449,10 +449,10 @@ class LoadImagesAndLabels(Dataset):
cache_path = (p if p.is_file() else Path(self.label_files[0]).parent).with_suffix('.cache')
try:
cache, exists = np.load(cache_path, allow_pickle=True).item(), True # load dict
assert cache['version'] == self.cache_version # same version
assert cache['hash'] == get_hash(self.label_files + self.im_files) # same hash
assert cache['version'] == self.cache_version # matches current version
assert cache['hash'] == get_hash(self.label_files + self.im_files) # identical hash
except Exception:
cache, exists = self.cache_labels(cache_path, prefix), False # cache
cache, exists = self.cache_labels(cache_path, prefix), False # run cache ops

# Display cache
nf, nm, ne, nc, n = cache.pop('results') # found, missing, empty, corrupt, total

Loading…
Cancel
Save