p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix bug v0.7

This commit is contained in:
xinjun ma 2021-06-22 21:48:21 +08:00
parent 8b5fff517d
commit 612018c4ad
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
model_zoo/research/cv/eppmvsnet/src/eppmvsnet.py
View File

@ -472,7 +472,7 @@ class EPPMVSNetP3(nn.Cell):
depth_interval_override=stage2_conf_interval,
uncertainty_maps=uncertainty_maps_2)
stage3_conf_interval = self.shrink_ratio * conf_range_2 / self.n_depths[1] * (
stage2_conf_interval * self.n_depths[1]) / self.n_depths[2]
stage2_conf_interval * self.n_depths[1]) / self.n_depths[2]
else:
est_depth_2, _, _ = self.stage2([ref_feat_2, srcs_feat_2, proj_mats[:, :, 1]],
depth_num=self.n_depths[1], depth_start_override=depth_start_2,
@ -496,8 +496,7 @@ class EPPMVSNetP3(nn.Cell):
est_depth_3, prob_map_3, _ = self.stage3([ref_feat_3, srcs_feat_3, proj_mats[:, :, 0]],
depth_num=self.n_depths[2],
depth_start_override=depth_start_3,
depth_interval_override=depth_interval *
self.interval_ratios[2],
depth_interval_override=depth_interval * self.interval_ratios[2],
uncertainty_maps=uncertainty_maps_3)
refined_depth = est_depth_3
return refined_depth, prob_map_3