Commit 59868eb
committed
refactor(deflow): update deflow code to the latest.
* faster, easier setup, better results for updated weight.
* eval script auto make also.
* cmakelists.txt add comment force set python executable.1 parent df8eae5 commit 59868eb
3 files changed
Lines changed: 12 additions & 5 deletions
- .github/issue_stale.yaml+16
- .gitignore+10-1
- .gitmodules-3
- Dockerfile+41
- README.md+35-15
- assets/README.md+63-13
- assets/cuda/README.md+21
- assets/cuda/chamfer3D/README.md+155
- assets/cuda/chamfer3D/__init__.py+117
- assets/cuda/chamfer3D/chamfer3D.cu+148
- assets/cuda/chamfer3D/chamfer3D_cuda.cpp+35
- assets/cuda/chamfer3D/setup.py+15
- assets/cuda/mmcv/README.md+4
- assets/cuda/mmcv/__init__.py+8
- assets/cuda/mmcv/common_cuda_helper.hpp+120
- assets/cuda/mmcv/cudabind.cpp+104
- assets/cuda/mmcv/pybind.cpp+50
- assets/cuda/mmcv/pytorch_cpp_helper.hpp+27
- assets/cuda/mmcv/pytorch_cuda_helper.hpp+20
- assets/cuda/mmcv/pytorch_device_registry.hpp+141
- assets/cuda/mmcv/scatter_points.cpp+53
- assets/cuda/mmcv/scatter_points.py+154
- assets/cuda/mmcv/scatter_points_cuda.cu+132
- assets/cuda/mmcv/scatter_points_cuda_kernel.cuh+187
- assets/cuda/mmcv/setup.py+52
- assets/cuda/mmcv/voxelization.cpp+74
- assets/cuda/mmcv/voxelization_cuda.cu+286
- assets/cuda/mmcv/voxelization_cuda_kernel.cuh+216
- assets/cuda/mmcv/voxelize.py+189
- assets/docs/index_eval_v2.pkl
- assets/dynamic_benchmark.png
- assets/slurm/0_process.sh+12-12
- assets/slurm/1_train.sh+13-13
- assets/slurm/2_eval.sh+4-4
- assets/slurm/dufolabel_sbatch.py+58
- assets/tests/README.md+42
- assets/tests/chamferdis_speed_test.py+128
- assets/tests/hdbscan_speed.py+99
- assets/tests/test_pc0.npy
- assets/tests/test_pc1.npy
- assets/view/av2.json+5-5
- conf/dynamicmap.yaml+11-11
- conf/hydra/default.yaml+1-1
- conf/model/deflow.yaml+2-2
- conf/save.yaml+13
- dataprocess/README.md+153
- dataprocess/extract_nus.py
- dataprocess/extract_waymo.py+425
- dataprocess/misc_data.py+91
- environment.yaml+10-4
- logs/.gitkeep
- main.py+19-11
- mmcv-1
- scripts/network/loss_func.py-133
- scripts/network/models/basic/__init__.py-14
- scripts/network/models/basic/embedder_model.py-73
- scripts/network/models/basic/fast_flow_decoder.py-227
- scripts/network/models/basic/feature_pyramid.py-157
- scripts/network/models/basic/make_voxels.py-89
- scripts/network/models/basic/scatter.py-94
- scripts/network/models/nsfp.py-141
- scripts/utils/o3d_view.py-119
- src/dataset.py+110-54
- src/lossfuncs.py+157
- src/models/__init__.py+2
- src/models/basic/__init__.py+35
- src/models/basic/decoder.py+27-5
- src/models/basic/encoder.py+222-50
- src/models/basic/nsfp_module.py
- src/models/basic/unet.py+1-21
- src/models/deflow.py+5-4
- src/models/fastflow3d.py+5-10
- src/trainer.py+165-101
- src/utils/__init__.py
- src/utils/av2_eval.py+201-16
- src/utils/eval_metric.py+272
- src/utils/mics.py+45-38
- src/utils/o3d_view.py+145
- src/utils/pcdpy3.py
- tests/README.md-16
- tests/scene_flow.py-92
- tests/submit_online.py-121
- tools/README.md+36
- tools/visualization.py+118
- tools/write4conf.py+23
- tools/zerof2ours.py+30
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
17 | 17 | | |
18 | 18 | | |
19 | 19 | | |
| 20 | + | |
| 21 | + | |
| 22 | + | |
20 | 23 | | |
21 | 24 | | |
22 | 25 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
1 | | - | |
2 | | - | |
| 1 | + | |
| 2 | + | |
3 | 3 | | |
4 | 4 | | |
5 | 5 | | |
6 | 6 | | |
7 | 7 | | |
8 | 8 | | |
9 | | - | |
10 | | - | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
11 | 15 | | |
12 | 16 | | |
13 | 17 | | |
0 commit comments