d4l3k / torchdrive

torch models and primitives for birdseye view and voxel models for self driving tasks.
https://fn.lc/post/3d-detr/
BSD 3-Clause "New" or "Revised" License
81 stars 7 forks source link

Share sample dataset #7

Open datvuthanh opened 1 year ago

datvuthanh commented 1 year ago

Hi Rice,

  1. Can you share one sample data (including image, mask and camera's calibration)? I cannot debug your source code.
  2. How can I generate mask for each camera?
a = MultiCamDataset(
    index_file="../../openape/snapshots/out-mar23/index.txt",
    mask_dir="../../openape/masks",
    cameras=["main", "narrow", "fisheye", "leftpillar", "leftrepeater", "rightpillar", "rightrepeater", "backup"],
    cam_shape=(480, 640),
    nframes_per_point=5,
    limit_size=1000,
    dynamic=True,
)

Thank you very much!