• Darijan Gudelj's avatar
    Heterogeneous raysampling -> RayBundleHeterogeneous · 6ae863f3
    Darijan Gudelj authored
    Summary:
    Added heterogeneous raysampling to pytorch3d raysampler, different cameras are sampled different number of times.
    
     It now returns RayBundle if heterogeneous raysampling is off and new RayBundleHeterogeneous (with added fields `camera_ids` and `camera_counts`).  Heterogeneous raysampling is on if `n_rays_total` is not None.
    
    Reviewed By: bottler
    
    Differential Revision: D39542222
    
    fbshipit-source-id: d3d88d822ec7696e856007c088dc36a1cfa8c625
    6ae863f3
test_raysampling.py 22.3 KB