sangramrout commited on
Commit
21fb724
·
verified ·
1 Parent(s): 5b84c16

Add model card

Browse files
Files changed (1) hide show
  1. README.md +55 -0
README.md ADDED
@@ -0,0 +1,55 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - lidar
5
+ - diffusion
6
+ - autonomous-driving
7
+ - sensor-synthesis
8
+ - nuscenes
9
+ - dinov3
10
+ library_name: pytorch
11
+ ---
12
+
13
+ # Sensor2Sensor — Camera → LiDAR synthesis checkpoints
14
+
15
+ Pretrained weights for **cross-modal sensor synthesis** on nuScenes: given surround
16
+ camera images, generate the corresponding LiDAR point cloud via a
17
+ **DINOv3-conditioned latent diffusion** model over a compact **LiDAR range-image VAE**.
18
+
19
+ - **Code / docs:** https://github.com/skr3178/sensor2sensor
20
+ - **Trained on:** a single RTX 3060 (11.6 GB), nuScenes v1.0-trainval.
21
+ - **Scope:** architecture validation on a small compute budget, not paper-level quality.
22
+
23
+ ## Files
24
+
25
+ | File | Model | Params | Size |
26
+ |---|---|---|---|
27
+ | `lidar_vae_best.pt` | v5 LiDAR range-image VAE (encoder μ + decoder) | ~2.07 M | 8.3 MB |
28
+ | `lidar_unet_best.pt` | 850-scenes DINOv3-conditioned diffusion U-Net (best held-out CD) | — | 59 MB |
29
+ | `lidar_unet_ema.pt` | EMA weights of the diffusion U-Net | — | 59 MB |
30
+
31
+ ## Held-out metrics (Chamfer distance, metres, lower = better; cfg=3.5, DDIM-25)
32
+
33
+ | Component | Metric | Value |
34
+ |---|---|---|
35
+ | LiDAR VAE (v5) | `CD-VAE-only` (decode(μ) vs raw) | **0.791 m** |
36
+ | Diffusion U-Net (850-scenes) | `CD-3D-raw` (N=16 held-out) | **1.994 m** |
37
+ | Diffusion U-Net (850-scenes) | `CD-BEV` (N=16 held-out) | **1.220 m** |
38
+ | End-to-end (VAE + diffusion) | `CD-3D-raw` (4 held-out keyframes) | **3.036 m** |
39
+
40
+ The 850-scenes checkpoint is selected by **held-out Chamfer distance measured in-loop**,
41
+ not by training MSE (a checkpoint with lower training MSE generalized worse). See the
42
+ repo's `s2s_min/RESULTS.md` §15 for the full rationale.
43
+
44
+ ## Loading
45
+
46
+ ```python
47
+ import torch
48
+ ckpt = torch.load("lidar_vae_best.pt", map_location="cuda")
49
+ # state dict keyed by the training-time best l1_range_ema basin; see the GitHub repo
50
+ # (s2s_min/models/) for the matching module definitions.
51
+ ```
52
+
53
+ ## License
54
+
55
+ MIT (see the GitHub repository). nuScenes data is subject to its own license/terms.