├── .gitignore ├── LICENSE ├── README.md ├── assets ├── 8nkt6-f4koq.gif ├── demo.gif ├── details.png └── prediction.gif ├── data_converter.py ├── project ├── builder │ ├── __init__.py │ ├── data_builder.py │ ├── loss_builder.py │ └── occ_occupancy_builder.py ├── config │ ├── __init__.py │ ├── _base_ │ │ ├── dataset.py │ │ ├── optimizer.py │ │ └── schedule.py │ ├── label_mapping │ │ ├── nuscenes-noIgnore.yaml │ │ └── nuscenes.yaml │ └── occupancy.py ├── dataloader │ ├── __init__.py │ ├── dataset.py │ ├── dataset_wrapper.py │ └── transform_3d.py ├── eval.py ├── launcher.sh ├── occupancy │ ├── __init__.py │ ├── dense_heads │ │ ├── Occformer_head.py │ │ └── __init__.py │ ├── detectors │ │ ├── __init__.py │ │ ├── grid_mask.py │ │ ├── occ_aggregator.py │ │ └── occformer.py │ └── modules │ │ ├── __init__.py │ │ ├── custom_base_transformer_layer.py │ │ ├── encoder.py │ │ ├── multi_scale_deformable_attn_function.py │ │ ├── spatial_cross_attention.py │ │ ├── temporal_self_attention.py │ │ └── transformer.py ├── requirements.txt ├── train.py └── utils │ ├── __init__.py │ ├── load_save_util.py │ ├── lovasz_losses.py │ ├── metric_util.py │ └── vis_image.py └── utils ├── points_process.py └── vis_pts.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/README.md -------------------------------------------------------------------------------- /assets/8nkt6-f4koq.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/assets/8nkt6-f4koq.gif -------------------------------------------------------------------------------- /assets/demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/assets/demo.gif -------------------------------------------------------------------------------- /assets/details.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/assets/details.png -------------------------------------------------------------------------------- /assets/prediction.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/assets/prediction.gif -------------------------------------------------------------------------------- /data_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/data_converter.py -------------------------------------------------------------------------------- /project/builder/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /project/builder/data_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/builder/data_builder.py -------------------------------------------------------------------------------- /project/builder/loss_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/builder/loss_builder.py -------------------------------------------------------------------------------- /project/builder/occ_occupancy_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/builder/occ_occupancy_builder.py -------------------------------------------------------------------------------- /project/config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /project/config/_base_/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/config/_base_/dataset.py -------------------------------------------------------------------------------- /project/config/_base_/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/config/_base_/optimizer.py -------------------------------------------------------------------------------- /project/config/_base_/schedule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/config/_base_/schedule.py -------------------------------------------------------------------------------- /project/config/label_mapping/nuscenes-noIgnore.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/config/label_mapping/nuscenes-noIgnore.yaml -------------------------------------------------------------------------------- /project/config/label_mapping/nuscenes.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/config/label_mapping/nuscenes.yaml -------------------------------------------------------------------------------- /project/config/occupancy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/config/occupancy.py -------------------------------------------------------------------------------- /project/dataloader/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /project/dataloader/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/dataloader/dataset.py -------------------------------------------------------------------------------- /project/dataloader/dataset_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/dataloader/dataset_wrapper.py -------------------------------------------------------------------------------- /project/dataloader/transform_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/dataloader/transform_3d.py -------------------------------------------------------------------------------- /project/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/eval.py -------------------------------------------------------------------------------- /project/launcher.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/launcher.sh -------------------------------------------------------------------------------- /project/occupancy/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/occupancy/__init__.py -------------------------------------------------------------------------------- /project/occupancy/dense_heads/Occformer_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/occupancy/dense_heads/Occformer_head.py -------------------------------------------------------------------------------- /project/occupancy/dense_heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/occupancy/dense_heads/__init__.py -------------------------------------------------------------------------------- /project/occupancy/detectors/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/occupancy/detectors/__init__.py -------------------------------------------------------------------------------- /project/occupancy/detectors/grid_mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/occupancy/detectors/grid_mask.py -------------------------------------------------------------------------------- /project/occupancy/detectors/occ_aggregator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/occupancy/detectors/occ_aggregator.py -------------------------------------------------------------------------------- /project/occupancy/detectors/occformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/occupancy/detectors/occformer.py -------------------------------------------------------------------------------- /project/occupancy/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/occupancy/modules/__init__.py -------------------------------------------------------------------------------- /project/occupancy/modules/custom_base_transformer_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/occupancy/modules/custom_base_transformer_layer.py -------------------------------------------------------------------------------- /project/occupancy/modules/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/occupancy/modules/encoder.py -------------------------------------------------------------------------------- /project/occupancy/modules/multi_scale_deformable_attn_function.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/occupancy/modules/multi_scale_deformable_attn_function.py -------------------------------------------------------------------------------- /project/occupancy/modules/spatial_cross_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/occupancy/modules/spatial_cross_attention.py -------------------------------------------------------------------------------- /project/occupancy/modules/temporal_self_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/occupancy/modules/temporal_self_attention.py -------------------------------------------------------------------------------- /project/occupancy/modules/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/occupancy/modules/transformer.py -------------------------------------------------------------------------------- /project/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/requirements.txt -------------------------------------------------------------------------------- /project/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/train.py -------------------------------------------------------------------------------- /project/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /project/utils/load_save_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/utils/load_save_util.py -------------------------------------------------------------------------------- /project/utils/lovasz_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/utils/lovasz_losses.py -------------------------------------------------------------------------------- /project/utils/metric_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/utils/metric_util.py -------------------------------------------------------------------------------- /project/utils/vis_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/project/utils/vis_image.py -------------------------------------------------------------------------------- /utils/points_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/utils/points_process.py -------------------------------------------------------------------------------- /utils/vis_pts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FANG-MING/occupancy-for-nuscenes/HEAD/utils/vis_pts.py --------------------------------------------------------------------------------