PistonY / torch-toolbox

🛠 Toolbox to extend PyTorch functionalities
BSD 3-Clause "New" or "Revised" License
417 stars 56 forks source link

ImageLMDB subdir error #3

Closed sailfish009 closed 4 years ago

sailfish009 commented 4 years ago

set subdir=False, otherwise, subdir=os.path.isdir(db_path)

https://github.com/PistonY/torch-toolbox/compare/master...sailfish009:master


.local/lib/python3.7/site-packages/torchtoolbox/data/lmdb_dataset.py", line 23, in __init__
    readahead=False, meminit=False)
lmdb.Error: /tmp/train_lmdb/train_image.lmdb: Not a directory
PistonY commented 4 years ago

Fix in master. https://github.com/PistonY/torch-toolbox/blob/c9070e6e643207711ad9200ae34d239fedc7e828/torchtoolbox/data/lmdb_dataset.py#L19