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

fix bug v0.3

This commit is contained in:
xinjun ma 2021-06-22 20:23:36 +08:00
parent 169b4b587e
commit 95f0a855ec
2 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/research/cv/eppmvsnet/src/blendedmvs.py
View File

@ -195,7 +195,7 @@ class BlendedMVSDataset:
masks = {"level_0": depth_0 > depth_min,
"level_1": depth_1 > depth_min,
"level_2": depth_2 > depth_min, }
"level_2": depth_2 > depth_min}
depth_max = depth_0.max()
return depths, masks, depth_max

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

@ -489,16 +489,16 @@ class EPPMVSNetP3(nn.Cell):
if self.entropy_range:
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=stage3_conf_interval,
uncertainty_maps=uncertainty_maps_3)
depth_num=self.n_depths[2],
depth_start_override=depth_start_3,
depth_interval_override=stage3_conf_interval,
uncertainty_maps=uncertainty_maps_3)
else:
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],
uncertainty_maps=uncertainty_maps_3)
depth_num=self.n_depths[2],
depth_start_override=depth_start_3,
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