Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
ioeddk authored Nov 12, 2024
1 parent f6fe3f1 commit 20ab2e2
Showing 1 changed file with 1 addition and 6 deletions.
7 changes: 1 addition & 6 deletions examples/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -19,11 +19,6 @@ The dimensions of TorchSparse differs from the SpConv, so the parameter dimensio
- `framework`: choose between `mmdet3d` and `openpc`.

## Example Conversion Commands
parser.add_argument("--ckpt_before", help="Path to the SpConv checkpoint")
parser.add_argument("--ckpt_after", help="Path to the output folder of the converted checkpoint.")
parser.add_argument("--cfg_path", help="Path to the config file of the model")
parser.add_argument("--v_spconv", default="1", help="SpConv version used for the weights. Can be one of 1 or 2")
parser.add_argument("--framework", default="mmdet3d", help="From which framework does the model weight comes from, choose one of mmdet3d or openpc")
### MMDetection3D
```bash
python examples/converter.py --ckpt_before ../mmdetection3d/models/PV-RCNN/pv_rcnn_8xb2-80e_kitti-3d-3class_20221117_234428-b384d22f.pth --cfg_path ../mmdetection3d/pv_rcnn/pv_rcnn_8xb2-80e_kitti-3d-3class.py --ckpt_after ./converted/PV-RCNN/pv_rcnn_8xb2-80e_kitti-3d-3class_20221117_234428-b384d22f.pth --v_spconv 1 --framework mmdet3d
Expand All @@ -38,4 +33,4 @@ python examples/converter.py --ckpt_before ../OpenPCDet/models/SECOND/second_786
Use the `test.py` that comes with the MMDet3D or OPC to run the evaluation. Provide the converted checkpoint as the model weights. For MMDet3D models, you need to provide extra arguments to replace certain layers to be torchsparse's (see how to replace them in `examples/mmdetection3d/demo.ipynb`). For OpenPCDet, the config file with those layers replaced is in the `examples/openpcdet/cfgs`; to use them, see `examples/openpcdet/demo.ipynb`. An additional step is to add `import ts_plugin` in `mmdetection3d/tools/test.py` and add `import pcdet_plugin` in `OpenPCDet/tools/test.py` to activate the plugins before running the evaluation.

# Details
Please see `examples/mmdetection3d/demo.ipynb` and `examples/openpcdet/demo.ipynb` for more details.
Please see `examples/mmdetection3d/demo.ipynb` and `examples/openpcdet/demo.ipynb` for more details.

0 comments on commit 20ab2e2

Please sign in to comment.