Skip to content

train on ETH3D dataset #70

Description

@wangtao-hku

Hi,

I downloaded the ETH3D dataset from MVSNET repo and use 'convet_eth3d_dataset.py' to process it.

However, i encountered an error during the training process.

Traceback (most recent call last):
File "c:\Users\SmartLab\Desktop\dl_recon_semantics\codes\Recon\PatchmatchNet-main\train.py", line 313, in
train(input_args, pmnet_model, train_loader, test_loader, pmnet_optimizer, epoch_start)
File "c:\Users\SmartLab\Desktop\dl_recon_semantics\codes\Recon\PatchmatchNet-main\train.py", line 41, in train
process_samples(args, train_sample, "train", logger, model, train_image_loader, optimizer, epoch_idx)
File "c:\Users\SmartLab\Desktop\dl_recon_semantics\codes\Recon\PatchmatchNet-main\train.py", line 92, in process_samples
for batch_idx, sample in enumerate(image_loader):
File "C:\Users\SmartLab\anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 652, in next
data = self._next_data()
File "C:\Users\SmartLab\anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 1347, in _next_data
return self._process_data(data)
File "C:\Users\SmartLab\anaconda3\lib\site-packages\torch\utils\data\dataloader.py", line 1373, in _process_data
data.reraise()
File "C:\Users\SmartLab\anaconda3\lib\site-packages\torch_utils.py", line 461, in reraise
raise exception
ValueError: Caught ValueError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "C:\Users\SmartLab\anaconda3\lib\site-packages\torch\utils\data_utils\worker.py", line 302, in _worker_loop
data = fetcher.fetch(index)
File "C:\Users\SmartLab\anaconda3\lib\site-packages\torch\utils\data_utils\fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "C:\Users\SmartLab\anaconda3\lib\site-packages\torch\utils\data_utils\fetch.py", line 49, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "c:\Users\SmartLab\Desktop\dl_recon_semantics\codes\Recon\PatchmatchNet-main\datasets\mvs.py", line 95, in getitem
depth_gt = read_map(depth_gt_filename, self.max_dim).transpose([2, 0, 1]).copy()
ValueError: axes don't match array

any help is much appreciated.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions