71 lines
1.8 KiB
Python
71 lines
1.8 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
简单测试Phase 4B配置是否能正常构建模型
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
sys.path.insert(0, '/workspace/bevfusion')
|
||
|
||
def test_config():
|
||
"""测试配置能否正常构建模型"""
|
||
try:
|
||
# 设置环境
|
||
os.environ['PYTHONPATH'] = '/workspace/bevfusion'
|
||
|
||
# 导入必要的模块
|
||
from mmcv import Config
|
||
from mmdet3d.models import build_model
|
||
|
||
print("🔧 加载Phase 4B配置...")
|
||
|
||
# 加载配置
|
||
config_file = 'configs/nuscenes/det/transfusion/secfpn/camera+lidar/swint_v0p075/multitask_BEV2X_phase4b_rmtppad_segmentation.yaml'
|
||
cfg = Config.fromfile(config_file)
|
||
|
||
print("✅ 配置加载成功")
|
||
|
||
# 简化配置用于测试
|
||
cfg.data.samples_per_gpu = 1
|
||
cfg.data.workers_per_gpu = 0
|
||
|
||
# 设置训练模式
|
||
cfg.model.train_cfg = {
|
||
'object': {
|
||
'grid_size': [1440, 1440, 41]
|
||
},
|
||
'map': {}
|
||
}
|
||
|
||
print("🔧 构建模型...")
|
||
|
||
# 构建模型
|
||
model = build_model(cfg.model, train_cfg=cfg.model.train_cfg)
|
||
|
||
print("✅ 模型构建成功")
|
||
print(f" - 模型类型: {type(model).__name__}")
|
||
|
||
# 检查heads
|
||
if hasattr(model, 'heads'):
|
||
heads = list(model.heads.keys())
|
||
print(f" - 模型heads: {heads}")
|
||
|
||
if 'map' in model.heads:
|
||
map_head = model.heads['map']
|
||
print(f" - 分割头类型: {type(map_head).__name__}")
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"❌ 测试失败: {e}")
|
||
import traceback
|
||
traceback.print_exc()
|
||
return False
|
||
|
||
if __name__ == "__main__":
|
||
success = test_config()
|
||
if success:
|
||
print("\n🎉 Phase 4B配置测试通过！")
|
||
else:
|
||
print("\n❌ 配置测试失败")
|