Convert image folder to lmdb, adapted from https://github.com/Lyken17/Efficient-PyTorch
.
βββ folder2lmdb.py
βββ img
βΒ Β βββ train
βΒ Β βΒ Β βββ bar_dir
βΒ Β βΒ Β βΒ Β βββ 100000.jpg
βΒ Β βΒ Β βΒ Β βββ 100001.jpg
βΒ Β βΒ Β βΒ Β βββ 100002.jpg
βΒ Β βΒ Β βΒ Β βββ 100003.jpg
βΒ Β βΒ Β βΒ Β βββ 100004.jpg
βΒ Β βΒ Β βΒ Β βββ 100005.jpg
βΒ Β βΒ Β βΒ Β βββ 100006.jpg
βΒ Β βΒ Β βΒ Β βββ 100007.jpg
βΒ Β βΒ Β βΒ Β βββ 100008.jpg
βΒ Β βΒ Β βΒ Β βββ 100009.jpg
βΒ Β βΒ Β βββ foo_dir
βΒ Β βΒ Β βββ 100000.jpg
βΒ Β βΒ Β βββ 100001.jpg
βΒ Β βΒ Β βββ 100002.jpg
βΒ Β βΒ Β βββ 100003.jpg
βΒ Β βΒ Β βββ 100004.jpg
βΒ Β βΒ Β βββ 100005.jpg
βΒ Β βΒ Β βββ 100006.jpg
βΒ Β βΒ Β βββ 100007.jpg
βΒ Β βΒ Β βββ 100008.jpg
βΒ Β βΒ Β βββ 100009.jpg
βΒ Β
βΒ Β
βββ main.py
βββ README.md
βββ requirements.txt
python folder2lmdb.py imgpython main.py img/train.lmdbkey 0
key 1
torch.Size([2, 224, 224, 3])
key 2
key 3
torch.Size([2, 224, 224, 3])
key 4
key 5
torch.Size([2, 224, 224, 3])
key 6
key 7
torch.Size([2, 224, 224, 3])
key 8
key 9
torch.Size([2, 224, 224, 3])
key 10
key 11
torch.Size([2, 224, 224, 3])
key 12
key 13
torch.Size([2, 224, 224, 3])
key 14
key 15
torch.Size([2, 224, 224, 3])
key 16
key 17
torch.Size([2, 224, 224, 3])
key 18
key 19
torch.Size([2, 224, 224, 3])