dontLoveBugs / DORN_pytorch

PyTorch implementation of Deep Ordinal Regression Network for Monocular Depth Estimation
299 stars 67 forks source link

modification in OrdinalRegressionLayer? #11

Closed jiaxinxie97 closed 4 years ago

jiaxinxie97 commented 5 years ago

https://github.com/dontLoveBugs/DORN_pytorch/blob/0e982db5ffca4c0c947f3c40baed5a065276e19b/network/DORN_kitti.py#L217 In line271 DORN_kitti.py, I find you output decode_c by summing all the probability. But in paper it counts channel with probability>0.5, your comments show similar operation in the paper. Can you tell me why?

dontLoveBugs commented 5 years ago

It seems a mistake. Thanks, I correct it.

jiaxinxie97 commented 5 years ago

@dontLoveBugs https://github.com/dontLoveBugs/DORN_pytorch/blob/1836f824bea6866e05147e6715024559f25607b4/dataloaders/kitti_dataloader.py#L111 Another likely bug, you change ground truth depth by multiplying s? I think it only need divide 100.

dontLoveBugs commented 5 years ago

s is scale factor, which is to zoom in the image and its groundtruth.

jiaxinxie97 commented 5 years ago

Oh,thank you. But I am still confused about why change groundtruth with a random number. If you do that, how to guarantee your output in a right scale? Or you mean because you zoom in the image so the groundtruth depth changes, you need to zoom it?

swethavsd commented 5 years ago

What is the size of the predicted ord labels from the Ordinal regression Layer? is it 1x71xHxW?