ialhashim / DenseDepth

High Quality Monocular Depth Estimation via Transfer Learning
https://arxiv.org/abs/1812.11941
GNU General Public License v3.0
1.58k stars 354 forks source link

Questions about the loss function of the Pytorch #221

Open liuyeqi94 opened 1 year ago

liuyeqi94 commented 1 year ago

Thanks a lot for making such an awesome research. I found some problems in running the code and would like to ask you for advice. In the Pytorch version of the code, the loss function in train.py consists of two parts, L1loss and SSIM, but the loss function in the paper consists of three parts, L1 loss, the L1 loss defined over the image gradient g of the depth image, and SSIM loss. Why is the second loss function missing from the loss function in the code? Is the impact significant? I am confused. Thanks again for your contribution and I am looking forward to your reply.