pytorch/vision

LSUN bedroom_train data load error

Open

#2,746 建立於 2020年10月2日

在 GitHub 查看
 (7 留言) (0 反應) (0 負責人)Python (15,050 star) (6,858 fork)batch import
help wantedmodule: datasetsnew feature

描述

🐛 Bug

I am trying to load the lsun bedroom_train data from torchvision.dataset but it's returning an error.

To Reproduce

Steps to reproduce the behavior:

  1. import torch
  2. from torchvision import transforms
  3. from torchvision import datasets
  4. import os
  5. path = os.getcwd()
  6. train_transform = transforms.Compose([
  7. transforms.ToTensor(),
  8. transforms.Normalize((0.5,0.5,0.5), (0.5, 0.5, 0.5))])
  9. bedroom = datasets.LSUN(root = path, classes= ['bedroom_train'], transform=train_transform)

Error Traceback (most recent call last) in ----> 1 bedroom = datasets.LSUN(root = path, classes= ['bedroom_train'], transform=train_transform)

D:\Softwares\Anaconda\lib\site-packages\torchvision\datasets\lsun.py in init(self, root, classes, transform, target_transform) 75 self.dbs.append(LSUNClass( 76 root=root + '/' + c + '_lmdb', ---> 77 transform=transform)) 78 79 self.indices = []

D:\Softwares\Anaconda\lib\site-packages\torchvision\datasets\lsun.py in init(self, root, transform, target_transform) 17 18 self.env = lmdb.open(root, max_readers=1, readonly=True, lock=False, ---> 19 readahead=False, meminit=False) 20 with self.env.begin(write=False) as txn: 21 self.length = txn.stat()['entries']

Error: G:\DCGANs/bedroom_train_lmdb: No such file or directory

Expected behavior

The LSUN bedroom training dataset should be downloaded in specified path

Environment

Please copy and paste the output from our environment collection script (or fill out the checklist below manually).

You can get the script and run it with:

wget https://raw.githubusercontent.com/pytorch/pytorch/master/torch/utils/collect_env.py
# For security purposes, please check the contents of collect_env.py before running it.
python collect_env.py
  • PyTorch / torchvision Version (e.g., 1.0 / 0.4.0): 1.6.0/0.7.0
  • OS (e.g., Linux): Windows 10
  • How you installed PyTorch / torchvision (conda, pip, source): conda
  • Build command you used (if compiling from source):
  • Python version: 3.7.7
  • CUDA/cuDNN version: 10.2
  • GPU models and configuration: NVIDIA GTX 1060 6GB
  • Any other relevant information:

Additional context

cc @pmeier

貢獻者指南