Xingang-Pan commited on
Commit
a33f468
β€’
1 Parent(s): 6f44efe

Revise to be consistent with the paper

Browse files
Files changed (1) hide show
  1. viz/renderer.py +3 -3
viz/renderer.py CHANGED
@@ -379,14 +379,14 @@ class Renderer:
379
  distance < round(r1 / 512 * h))
380
  direction = direction / \
381
  (torch.linalg.norm(direction) + 1e-7)
382
- gridh = (relis-direction[1]) / (h-1) * 2 - 1
383
- gridw = (reljs-direction[0]) / (w-1) * 2 - 1
384
  grid = torch.stack(
385
  [gridw, gridh], dim=-1).unsqueeze(0).unsqueeze(0)
386
  target = F.grid_sample(
387
  feat_resize.float(), grid, align_corners=True).squeeze(2)
388
  loss_motion += F.l1_loss(
389
- feat_resize[:, :, relis, reljs], target.detach())
390
 
391
  loss = loss_motion
392
  if mask is not None:
 
379
  distance < round(r1 / 512 * h))
380
  direction = direction / \
381
  (torch.linalg.norm(direction) + 1e-7)
382
+ gridh = (relis+direction[1]) / (h-1) * 2 - 1
383
+ gridw = (reljs+direction[0]) / (w-1) * 2 - 1
384
  grid = torch.stack(
385
  [gridw, gridh], dim=-1).unsqueeze(0).unsqueeze(0)
386
  target = F.grid_sample(
387
  feat_resize.float(), grid, align_corners=True).squeeze(2)
388
  loss_motion += F.l1_loss(
389
+ feat_resize[:, :, relis, reljs].detach(), target)
390
 
391
  loss = loss_motion
392
  if mask is not None: