Some errors in the code

[gangweiX]

for cfnet.py, First error: def generate_search_range(self, sample_count, input_min_disparity, input_max_disparity): """ Description: Generates the disparity search range. Returns: :min_disparity: Lower bound of disparity search range :max_disparity: Upper bound of disaprity search range. """

    min_disparity = torch.clamp(input_min_disparity - torch.clamp((
            sample_count - input_max_disparity + input_min_disparity), min=0) / 2.0, min=0, max=self.maxdisp)
    max_disparity = torch.clamp(input_max_disparity + torch.clamp(
            sample_count - input_max_disparity + input_min_disparity, min=0) / 2.0, min=0, max=self.maxdisp)

    return min_disparity, max_disparity

it should be "min_disparity = torch.clamp(input_min_disparity - torch.clamp(( sample_count - input_max_disparity + input_min_disparity), min=0) / 2.0, min=0, max=self.maxdisp//4-1) max_disparity = torch.clamp(input_max_disparity + torch.clamp( sample_count - input_max_disparity + input_min_disparity, min=0) / 2.0, min=0, max=self.maxdisp//4-1)" or "min_disparity = torch.clamp(input_min_disparity - torch.clamp(( sample_count - input_max_disparity + input_min_disparity), min=0) / 2.0, min=0, max=self.maxdisp//2-1) max_disparity = torch.clamp(input_max_disparity + torch.clamp( sample_count - input_max_disparity + input_min_disparity, min=0) / 2.0, min=0, max=self.maxdisp//2-1)"

Second error: in line 643 of cfnet.py, it should be "predmid_s2 = F.upsample(predmid_s2 2, [left.size()[2], left.size()[3]], mode='bilinear', align_corners=True)", not "predmid_s2 = F.upsample(predmid_s2 4, [left.size()[2], left.size()[3]], mode='bilinear', align_corners=True)"

[gallenszl]

Thank you very much! I have fixed the two bugs in the cfnet.py file.