【MMDetection系列 - 1】 MMdetection基础环境配置及demo运行

### 运行 mmdetection3d Demo 并实现可视化的具体方法 要运行 mmdetection3d 的 demo 并实现可视化，可以按照以下方式操作： #### 准备工作 1. **下载预训练模型** 需要先从指定链接下载预训练模型文件，并将其保存到 `checkpoints` 文件夹中。例如，可以从以下地址获取模型文件[^2]： - 官方下载地址：https://download.openmmlab.com/mmdetection3d/v0.1.0_models/second/hv_second_secfpn_6x8_80e_kitti-3d-car/hv_second_secfpn_6x8_80e_kitti-3d-car_20200620_230238-393f000c.pth - 或者通过 CSDN 下载。 将下载好的 `.pth` 文件放置于项目根目录下的 `checkpoints` 文件夹中。 2. **准备输入数据** 输入的数据应为点云格式（如 `.bin` 或 `.pcd`）。如果需要处理自定义的点云文件，请确保其格式与配置文件一致。 3. **验证环境安装** 确认已正确安装所有依赖项，并能够正常调用 GPU 加速功能。可以通过以下命令测试环境是否搭建成功[^3]： ```python import torch print(torch.cuda.is_available()) # 输出 True 表明 CUDA 支持可用 ``` #### 执行 Demo 脚本 执行官方提供的 `pcd_demo.py` 脚本来完成检测和可视化任务。以下是具体的命令结构[^4]: ```bash python demo/pcd_demo.py \ <path_to_input_pcd_or_bin> \ <config_file_path> \ <checkpoint_file_path> \ [--out-dir <output_directory>] \ [--device <cpu/cuda:0>] \ [--score-thr <threshold_value>] \ [--show] ``` 其中各参数含义如下： - `<path_to_input_pcd_or_bin>`：待检测的点云文件路径。 - `<config_file_path>`：对应的配置文件路径，用于描述模型架构及相关超参设置。 - `<checkpoint_file_path>`：预训练权重文件路径。 - `[--out-dir <output_directory>]`：可选参数，指定输出结果存储位置。 - `[--device <cpu/cuda:0>]`：设备选项，默认使用 CPU；推荐显卡支持的情况下设为 `cuda:0`。 - `[--score-thr <threshold_value>]`：置信度阈值过滤条件，默认值通常为 0.5。 - `[--show]`：开启实时显示模式，在屏幕上展示预测框及其类别标签。 示例命令如下所示： ```bash python demo/pcd_demo.py \ /home/user/data/custom_point_cloud.bin \ configs/pointpillars/pointpillars_hv_secfpn_8xb6-160e_kitti-3d-car.py \ checkpoints/hv_pointpillars_secfpn_6x8_160e_kitti-3d-car_20220331_134606-d42d15ed.pth \ --show ``` 此脚本会读取输入点云数据，利用加载的模型进行推理运算，并最终呈现出带有标注边界的三维物体边界框效果。 #### 使用 MMEngine 可视化工具增强体验 除了上述基础流程外，还可以借助 MMEngine 中内置的 `Visualizer` 工具进一步提升交互性和灵活性[^1]。它允许开发者更加精细地控制哪些阶段的状态或中间产物被记录下来以便后续分析查看。 --- ###