mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fix bug v0.6

This commit is contained in:
xinjun ma 2021-06-22 21:06:53 +08:00
parent 5e3c3672b6
commit 8b5fff517d
1 changed files with 10 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -407,7 +407,8 @@ class EPPMVSNetP1(nn.Cell):
depth_num=self.n_depths[0],
depth_start_override=depth_start,
depth_interval_override=depth_interval *
self.interval_ratios[0])
self.interval_ratios[
0])
_, src_feats, proj_mats = [ref_feat_1, srcs_feat_1, proj_mats[:, :, 2]]
pixel_distances = []
@ -445,16 +446,14 @@ class EPPMVSNetP3(nn.Cell):
ref_feat_1, srcs_feat_1 = feat_pack_1[:, 0], feat_pack_1[:, 1:]
if self.entropy_range:
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,
proj_mats[:, :, 2]],
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, proj_mats[:, :, 2]],
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]
else:
est_depth_1, _, pair_results_1 = self.stage1_p3(cost_volume_list_stage1,
depth_values_stage1, [ref_feat_1, srcs_feat_1,
proj_mats[:, :, 2]],
est_depth_1, _, pair_results_1 = self.stage1_p3(cost_volume_list_stage1, depth_values_stage1,
[ref_feat_1, srcs_feat_1, proj_mats[:, :, 2]],
self.n_depths[0])
stage2_conf_interval = None
uncertainty_maps_1, uncertainty_maps_2 = [], []