wuwenhai
/
bev-project
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
bev-project/mmdet3d/ops/__init__.py

95 lines
2.3 KiB
Python
Raw Normal View History

[Major] Code release.
2022-06-03 12:21:18 +08:00
from mmcv.ops import (
RoIAlign,
SigmoidFocalLoss,
get_compiler_version,
get_compiling_cuda_version,
nms,
roi_align,
sigmoid_focal_loss,
)
from .ball_query import ball_query
Complete project state snapshot: Phase 4B RMT-PPAD Integration 🎯 Training Status: - Current Epoch: 2/10 (13.3% complete) - Segmentation Dice: 0.9594 - Detection IoU: 0.5742 - Training stable with 8 GPUs 🔧 Technical Achievements: - ✅ RMT-PPAD Transformer segmentation decoder integrated - ✅ Task-specific GCA architecture optimized - ✅ Multi-scale feature fusion (180×180, 360×360, 600×600) - ✅ Adaptive scale weight learning implemented - ✅ BEVFusion multi-task framework enhanced 📊 Performance Highlights: - Divider segmentation: 0.9793 Dice (excellent) - Pedestrian crossing: 0.9812 Dice (excellent) - Stop line: 0.9812 Dice (excellent) - Carpark area: 0.9802 Dice (excellent) - Walkway: 0.9401 Dice (good) - Drivable area: 0.8959 Dice (good) 🛠️ Code Changes Included: - Enhanced BEVFusion model (bevfusion.py) - RMT-PPAD integration modules (rmtppad_integration.py) - Transformer segmentation head (enhanced_transformer.py) - GCA module optimizations (gca.py) - Configuration updates (Phase 4B configs) - Training scripts and automation tools - Comprehensive documentation and analysis reports 📅 Snapshot Date: Fri Nov 14 09:06:09 UTC 2025 📍 Environment: Docker container 🎯 Phase: RMT-PPAD Integration Complete
2025-11-14 17:06:09 +08:00
#from .feature_decorator import feature_decorator
[Major] Code release.
2022-06-03 12:21:18 +08:00
from .furthest_point_sample import (
Points_Sampler,
furthest_point_sample,
furthest_point_sample_with_dist,
)
from .gather_points import gather_points
from .group_points import GroupAll, QueryAndGroup, group_points, grouping_operation
from .interpolate import three_interpolate, three_nn
from .knn import knn
from .norm import NaiveSyncBatchNorm1d, NaiveSyncBatchNorm2d
from .paconv import PAConv, PAConvCUDA, assign_score_withk
from .bev_pool import *
from .pointnet_modules import (
PAConvCUDASAModule,
PAConvCUDASAModuleMSG,
PAConvSAModule,
PAConvSAModuleMSG,
PointFPModule,
PointSAModule,
PointSAModuleMSG,
build_sa_module,
)
from .roiaware_pool3d import (
RoIAwarePool3d,
points_in_boxes_batch,
points_in_boxes_cpu,
points_in_boxes_gpu,
)
from .sparse_block import SparseBasicBlock, SparseBottleneck, make_sparse_convmodule
from .voxel import DynamicScatter, Voxelization, dynamic_scatter, voxelization
__all__ = [
"nms",
"soft_nms",
"RoIAlign",
"roi_align",
"get_compiler_version",
"get_compiling_cuda_version",
"NaiveSyncBatchNorm1d",
"NaiveSyncBatchNorm2d",
"batched_nms",
"Voxelization",
"voxelization",
"dynamic_scatter",
"DynamicScatter",
"sigmoid_focal_loss",
"SigmoidFocalLoss",
"SparseBasicBlock",
"SparseBottleneck",
"RoIAwarePool3d",
"points_in_boxes_gpu",
"points_in_boxes_cpu",
"make_sparse_convmodule",
"ball_query",
"knn",
"furthest_point_sample",
"furthest_point_sample_with_dist",
"three_interpolate",
"three_nn",
"gather_points",
"grouping_operation",
"group_points",
"GroupAll",
"QueryAndGroup",
"PointSAModule",
"PointSAModuleMSG",
"PointFPModule",
"points_in_boxes_batch",
"get_compiler_version",
"assign_score_withk",
"get_compiling_cuda_version",
"Points_Sampler",
"build_sa_module",
"PAConv",
"PAConvCUDA",
"PAConvSAModuleMSG",
"PAConvSAModule",
"PAConvCUDASAModule",
"PAConvCUDASAModuleMSG",
"bev_pool",
Add BEVFusion-R (#443)
2023-07-08 10:53:36 +08:00
"feature_decorator",
[Major] Code release.
2022-06-03 12:21:18 +08:00
]