pytorch_lmdb_imagenet
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
- Change Imagenet train set and val set to lmdb
python folder2lmdb.py
- Test the dataset
python dataset_test.py
- Train Imagenet
python train_imagenet.py /path/to/dataset/imagenet/ --lmdb