From 4f8f1d79c8b8e530495b5f183280bab99869e845 Mon Sep 17 00:00:00 2001 From: Yuang Peng Date: Wed, 22 Feb 2023 11:41:18 +0800 Subject: [PATCH] fix(data): read img bug in voc dataset (#1615) fix(data): read img bug in voc dataset --- yolox/data/datasets/voc.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/yolox/data/datasets/voc.py b/yolox/data/datasets/voc.py index 62b539d..bdacd80 100644 --- a/yolox/data/datasets/voc.py +++ b/yolox/data/datasets/voc.py @@ -189,8 +189,8 @@ class VOCDetection(CacheDataset): return img - @cache_read_img - def read_img(self, index, use_cache=True): + @cache_read_img(use_cache=True) + def read_img(self, index): return self.load_resized_img(index) def pull_item(self, index): -- GitLab