xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix bug v0.4

This commit is contained in:
xinjun ma 2021-06-22 20:53:30 +08:00
parent 95f0a855ec
commit bbdb4dc758
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -445,14 +445,14 @@ class EPPMVSNetP3(nn.Cell):
ref_feat_1, srcs_feat_1 = feat_pack_1[:, 0], feat_pack_1[:, 1:] ref_feat_1, srcs_feat_1 = feat_pack_1[:, 0], feat_pack_1[:, 1:]
if self.entropy_range: if self.entropy_range:
est_depth_1, _, _, conf_range_1 = self.stage1_p3(cost_volume_list_stage1, est_depth_1, _, pair_results_1, conf_range_1 = self.stage1_p3(cost_volume_list_stage1,
depth_values_stage1, [ref_feat_1, srcs_feat_1, depth_values_stage1, [ref_feat_1, srcs_feat_1,
proj_mats[:, :, 2]], proj_mats[:, :, 2]],
self.n_depths[0]) self.n_depths[0])
stage2_conf_interval = self.shrink_ratio * conf_range_1 / self.n_depths[0] * ( stage2_conf_interval = self.shrink_ratio * conf_range_1 / self.n_depths[0] * (
depth_interval * self.interval_ratios[0] * self.n_depths[0]) / self.n_depths[1] depth_interval * self.interval_ratios[0] * self.n_depths[0]) / self.n_depths[1]
else: else:
est_depth_1, _, _ = self.stage1_p3(cost_volume_list_stage1, est_depth_1, _, pair_results_1 = self.stage1_p3(cost_volume_list_stage1,
depth_values_stage1, [ref_feat_1, srcs_feat_1, depth_values_stage1, [ref_feat_1, srcs_feat_1,
proj_mats[:, :, 2]], proj_mats[:, :, 2]],
self.n_depths[0]) self.n_depths[0])