pytorch_lmdb_imagenet icon indicating copy to clipboard operation
pytorch_lmdb_imagenet copied to clipboard

Use lmdb to speed up imagenet dataset

pytorch_lmdb_imagenet

Convert ImageNet dataset to lmdb, adapted from https://github.com/Lyken17/Efficient-PyTorch

  1. Change Imagenet train set and val set to lmdb
python folder2lmdb.py
  1. Test the dataset
python dataset_test.py
  1. Train Imagenet
python train_imagenet.py /path/to/dataset/imagenet/  --lmdb