Skip to content

Instantly share code, notes, and snippets.

@Mason-McGough

Mason-McGough/calc_volume_weights.py

Last active June 25, 2021 18:34
Show Gist options
  • Save Mason-McGough/90713e10f1dd4ae92750f91cb73c7371 to your computer and use it in GitHub Desktop.
Save Mason-McGough/90713e10f1dd4ae92750f91cb73c7371 to your computer and use it in GitHub Desktop.
Download ZIP
GIRAFFE volume weights
Raw
calc_volume_weights.py
# adapted from https://github.com/autonomousvision/giraffe (MIT License)
weights = self.calc_volume_weights(di, ray_vector, sigma_sum)
feat_map = torch.sum(weights.unsqueeze(-1) * feat_weighted, dim=-2)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment