Browse Source
Unique *.cache filenames fix (#2134)
* fix #2121 * Update test.py * Update train.py * Update autoanchor.py * Update datasets.py * Update log_dataset.py * Update datasets.py Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>5.0
train255
GitHub
3 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
utils/datasets.py
View File
| # Check cache | # Check cache | ||||
| self.label_files = img2label_paths(self.img_files) # labels | self.label_files = img2label_paths(self.img_files) # labels | ||||
| cache_path = Path(self.label_files[0]).parent.with_suffix('.cache') # cached labels | |||||
| cache_path = (p if p.is_file() else Path(self.label_files[0]).parent).with_suffix('.cache') # cached labels | |||||
| if cache_path.is_file(): | if cache_path.is_file(): | ||||
| cache = torch.load(cache_path) # load | cache = torch.load(cache_path) # load | ||||
| if cache['hash'] != get_hash(self.label_files + self.img_files) or 'results' not in cache: # changed | if cache['hash'] != get_hash(self.label_files + self.img_files) or 'results' not in cache: # changed |
Loading…