Compute RGB MSE Loss, original from NeRF Paper
(rgb, target, extras)
| 73 | return pose |
| 74 | |
| 75 | def rgb_loss(rgb, target, extras): |
| 76 | ''' Compute RGB MSE Loss, original from NeRF Paper ''' |
| 77 | # Compute MSE loss between predicted and true RGB. |
| 78 | img_loss = img2mse(rgb, target) |
| 79 | loss = img_loss |
| 80 | |
| 81 | # Add MSE loss for coarse-grained model |
| 82 | if 'rgb0' in extras: |
| 83 | img_loss0 = img2mse(extras['rgb0'], target) |
| 84 | loss += img_loss0 |
| 85 | return loss |
| 86 | |
| 87 | def PoseLoss(args, pose_, pose, device): |
| 88 | loss_func = nn.MSELoss() |