TRI-ML/packnet-sfm

BUG: SemiSupCompletionModel do not flip back the output['inv_depths_rgbd']

chenjx1005 opened this issue · 1 comments

Bug appears when I train SemiSupCompletionModel. The DEPTH_PP_GT metric is much worse than DEPTH_GT of my re-trained model.

I check the traning code. In

output = flip_output(output)
, I found if flip is True, you do flip back the output['inv_depths'] but not output['inv_depths_rgbd']. see:
def flip_output(output):
, 'inv_depths_rgbd' key in not in the fliter_dict. As a result, it causes the model can not handle fliped input.

Thank you for pointing that out! And thank you for your PR, I will take a look and merge soon.