megvii-research / TransMVSNet

(CVPR 2022) TransMVSNet: Global Context-aware Multi-view Stereo Network with Transformers.
MIT License
262 stars 25 forks source link

运行test_dtu.sh报错 #34

Open QZZZZZYF opened 11 months ago

QZZZZZYF commented 11 months ago

argv: ['--dataset=general_eval', '--batch_size=1', '--testpath=test/dtu', '--testlist=lists/dtu/test.txt', '--loadckpt=checkpoints/model_dtu.ckpt', '--outdir=outputs/dtu_testing', '--numdepth=192', '--ndepths=48,32,8', '--depth_inter_r=4.0,1.0,0.5', '--interval_scale=1.06', '--filter_method=gipuma', '--fusibile_exe_path='] ################################ args ################################ model mvsnet <class 'str'>
dataset general_eval <class 'str'>
testpath test/dtu <class 'str'>
testpath_single_scene None <class 'NoneType'>
testlist lists/dtu/test.txt <class 'str'>
batch_size 1 <class 'int'>
numdepth 192 <class 'int'>
loadckpt checkpoints/model_dtu.ckpt <class 'str'>
outdir outputs/dtu_testing <class 'str'>
display False <class 'bool'>
share_cr False <class 'bool'>
ndepths 48,32,8 <class 'str'>
depth_inter_r 4.0,1.0,0.5 <class 'str'>
cr_base_chs 8,8,8 <class 'str'>
grad_method detach <class 'str'>
interval_scale 1.06 <class 'float'>
num_view 5 <class 'int'>
max_h 864 <class 'int'>
max_w 1152 <class 'int'>
fix_res False <class 'bool'>
num_worker 4 <class 'int'>
save_freq 20 <class 'int'>
filter_method gipuma <class 'str'>
conf 0.03 <class 'float'>
thres_view 5 <class 'int'>
fusibile_exe_path <class 'str'>
prob_threshold 0.01 <class 'float'>
disp_threshold 0.25 <class 'float'>
num_consistent 3.0 <class 'float'>
######################################################################## *****Interval_Scale** 1.06 dataset test metas: 49 interval_scale:{'scan1': 1.06} **netphs:[48, 32, 8], depth_intervals_ratio:[4.0, 1.0, 0.5], grad:detach, chs:[8, 8, 8]**** loading model checkpoints/model_dtu.ckpt Traceback (most recent call last): File "test.py", line 433, in save_depth(testlist) File "test.py", line 133, in save_depth save_scene_depth([scene]) File "test.py", line 162, in save_scene_depth outputs = model(sample_cuda["imgs"], sample_cuda["proj_matrices"], sample_cuda["depth_values"]) File "/home/enb/Software/anaconda3/envs/transmvsnet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl result = self.forward(*input, kwargs) File "/home/enb/Software/anaconda3/envs/transmvsnet/lib/python3.6/site-packages/torch/nn/parallel/data_parallel.py", line 159, in forward return self.module(*inputs[0], kwargs[0]) File "/home/enb/Software/anaconda3/envs/transmvsnet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl result = self.forward(*input, *kwargs) File "/media/enb/d246805e-89c3-40bf-95da-9f0d81ea7b05/home/enb/ZYF/TransMVSNet-master/models/TransMVSNet.py", line 161, in forward features.append(self.feature(img)) File "/home/enb/Software/anaconda3/envs/transmvsnet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl result = self.forward(input, kwargs) File "/media/enb/d246805e-89c3-40bf-95da-9f0d81ea7b05/home/enb/ZYF/TransMVSNet-master/models/module.py", line 411, in forward out = self.out1(intra_feat) File "/home/enb/Software/anaconda3/envs/transmvsnet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl result = self.forward(*input, *kwargs) File "/home/enb/Software/anaconda3/envs/transmvsnet/lib/python3.6/site-packages/torch/nn/modules/container.py", line 117, in forward input = module(input) File "/home/enb/Software/anaconda3/envs/transmvsnet/lib/python3.6/site-packages/torch/nn/modules/module.py", line 727, in _call_impl result = self.forward(input, **kwargs) File "/media/enb/d246805e-89c3-40bf-95da-9f0d81ea7b05/home/enb/ZYF/TransMVSNet-master/models/dcn.py", line 79, in forward mask=mask TypeError: deform_conv2d() got an unexpected keyword argument 'mask'

QZZZZZYF commented 11 months ago

python=3.6 pytroch=1.7.1 cuda=11.0

jinzhenmu commented 5 months ago

您解决了吗?