Skip to content

S
seminar-breakout
Commit c6de1746 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix ilsvrccaffelmdb

parent a027b8de
Hide whitespace changes
Inline Side-by-side
Showing with 13 additions and 14 deletions
tensorpack/dataflow/dataset/ilsvrc.py
View file @ c6de1746
...@@ -174,20 +174,19 @@ class ILSVRC12CaffeLMDB(DataFlow): ...@@ -174,20 +174,19 @@ class ILSVRC12CaffeLMDB(DataFlow):
return None return None
return [img.transpose(1, 2, 0), datum.label] return [img.transpose(1, 2, 0), datum.label]
with self._txn: if not self._shuffle:
if not self._shuffle: c = self._txn.cursor()
c = self._txn.cursor() while c.next():
while c.next(): k, v = c.item()
k, v = c.item() v = parse(k, v)
v = parse(k, v) if v: yield v
if v: yield v else:
else: s = self.size()
s = self.size() for i in range(s):
for i in range(s): k = self.rng.choice(self.keys)
k = self.rng.choice(self.keys) v = self._txn.get(k)
v = self._txn.get(k) v = parse(k, v)
v = parse(k, v) if v: yield v
if v: yield v
if __name__ == '__main__': if __name__ == '__main__':
meta = ILSVRCMeta() meta = ILSVRCMeta()
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment