search

imran3180 / depth-map-prediction

Pytorch Implementation of Depth Map Prediction from a Single Image using a Multi-Scale Deep Network
https://cs.nyu.edu/~deigen/depth/
105 stars 13 forks source link

How to use this project? #3

Open LIZ-CRYSTAL opened 4 years ago

LIZ-CRYSTAL commented 4 years ago

Excuse me, I have download your repo, but there is not a guide, so I don't know how to use this repo, which file should I run first? And where should I put the NYU dataset?

gwcrepo commented 9 months ago

this could help train_loader = torch.utils.data.DataLoader(NYUDataset( args.data + '/nyu_depth_v2_labeled.mat', 'training', rgb_transform = rgb_data_transforms, depth_transform = depth_data_transforms), batch_size = args.batch_size, shuffle = True, num_workers = 5)