From 9ae868364a2d98bd03cecb8ba8f6310c0d11b482 Mon Sep 17 00:00:00 2001 From: tkianai <49858938+tkianai@users.noreply.github.com> Date: Tue, 11 Aug 2020 02:23:48 +0800 Subject: [PATCH] the value should be a list to unpacking (#688) --- utils/datasets.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/datasets.py b/utils/datasets.py index 169a518e0697..7ed8f5a1d805 100755 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -457,7 +457,7 @@ def cache_labels(self, path='labels.cache'): l = np.zeros((0, 5), dtype=np.float32) x[img] = [l, shape] except Exception as e: - x[img] = None + x[img] = [None, None] print('WARNING: %s: %s' % (img, e)) x['hash'] = get_hash(self.label_files + self.img_files)