| 130 | cv2.imwrite(save_path_img, rendered_img.astype(np.uint8)[...,::-1]) |
| 131 | |
| 132 | class DeltaDepth(nn.Module): |
| 133 | def __init__(self,bs): |
| 134 | super(DeltaDepth, self).__init__() |
| 135 | self.delta_d = nn.Parameter(torch.zeros(bs,1),requires_grad=True) |
| 136 | def forward(self,depth): |
| 137 | return depth+self.delta_d |
| 138 | |
| 139 | class DeltaDepthBeta(nn.Module): |
| 140 | def __init__(self,bs): |