JeffWang987 / MVSTER

[ECCV 2022] MVSTER: Epipolar Transformer for Efficient Multi-View Stereo
MIT License
183 stars 13 forks source link

Fine tuning on other datasets #5

Open wyg-okk opened 1 year ago

wyg-okk commented 1 year ago

Has anyone fine-tuned the pretrained model(https://github.com/JeffWang987/MVSTER/releases/tag/dtu_ckpt) provided by the author?When I did so, I found that the author may have left out "The Auxiliary Branch makes Monocular depth Estimation" from the model.

I got the following errors:

RuntimeError: Error(s) in loading state_dict for MVS4net: Missing key(s) in state_dict: "mono_depth_decoder.convblocks.0.conv.weight", "mono_depth_decoder.convblocks.0.bn.weight", "mono_depth_decoder.convblocks.0.bn.bias", "mono_depth_decoder.convblocks.0.bn.running_mean", "mono_depth_decoder.convblocks.0.bn.running_var", "mono_depth_decoder.convblocks.1.conv.weight", "mono_depth_decoder.convblocks.1.bn.weight", "mono_depth_decoder.convblocks.1.bn.bias", "mono_depth_decoder.convblocks.1.bn.running_mean", "mono_depth_decoder.convblocks.1.bn.running_var", "mono_depth_decoder.convblocks.2.conv.weight", "mono_depth_decoder.convblocks.2.bn.weight", "mono_depth_decoder.convblocks.2.bn.bias", "mono_depth_decoder.convblocks.2.bn.running_mean", "mono_depth_decoder.convblocks.2.bn.running_var", "mono_depth_decoder.conv3x3.0.weight", "mono_depth_decoder.conv3x3.0.bias", "mono_depth_decoder.conv3x3.1.weight", "mono_depth_decoder.conv3x3.1.bias", "mono_depth_decoder.conv3x3.2.weight", "mono_depth_decoder.conv3x3.2.bias".