yuangpeng commited on
Commit
5c838b8
·
1 Parent(s): cba5002

fix(data): read img bug in voc dataset (#1615)

Browse files
Files changed (1) hide show
  1. yolox/data/datasets/voc.py +2 -2
yolox/data/datasets/voc.py CHANGED
@@ -189,8 +189,8 @@ class VOCDetection(CacheDataset):
189
 
190
  return img
191
 
192
- @cache_read_img
193
- def read_img(self, index, use_cache=True):
194
  return self.load_resized_img(index)
195
 
196
  def pull_item(self, index):
 
189
 
190
  return img
191
 
192
+ @cache_read_img(use_cache=True)
193
+ def read_img(self, index):
194
  return self.load_resized_img(index)
195
 
196
  def pull_item(self, index):