使用DSGN++训练自己的数据集

本文档介绍了使用DSGN++训练自定义双目图像数据集时遇到的问题及解决方案,包括通道匹配、数据预处理、相机内参调整等。同时探讨了Waymo数据集的使用,提出了从多角度视图估计深度的方法,并分享了数据转换和测试的经验。
摘要由CSDN通过智能技术生成

自己采集的数据

由于图像是:RGB+灰度双目图像,所以涉及到通道个数不匹配问题等等,从头到尾凭记忆记录一下:
1.修改config\dsgn.yaml文件
把验证集换为测试集,这样就可以运行test.py时生成的是test文件夹下的结果(多此一举是因为我的数据集没有label,而test数据集不需要label)

DATA_CONFIG: 
    _BASE_CONFIG_: ./configs/stereo/dataset_configs/kitti_dataset_fused.yaml
    later_flip: True
    DATA_SPLIT: {
   
        'train': train,
        'test': test                # val
    }
    INFO_PATH: {
   
        'train': [kitti_infos_train.pkl],
        'test': [kitti_infos_test.pkl],           # kitti_infos_val.pkl
    }

2.生成kitti_infos_test.pkl文件
这里由于数据集没有label需要注释掉一部分代码,在dataset/kitti/lidar_kitti_dataset.py文件中注释掉453~467行,只保留生成test部分

python -m pcdet.datasets.kitti.lidar_kitti_dataset create_kitti_infos

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值