Commit a0cfd6e4 authored by Gus-Guo's avatar Gus-Guo
Browse files

update README.md

parent 62f2f5e7
......@@ -7,7 +7,7 @@ As of now, it mainly consists of `PCDet` toolbox for 3D object detection from po
### What does `PCDet` toolbox do?
Note that we have upgrated `PCDet` from `v0.1` to `v0.2` with prety new structures to support various datasets and models.
Note that we have upgrated `PCDet` from `v0.1` to `v0.2` with pretty new structures to support various datasets and models.
`PCDet` is a general PyTorch-based codebase for 3D object detection from point cloud.
It currently supports multiple state-of-the-art 3D object detection methods with highly refactored codes for both one-stage and two-stage 3D detection frameworks.
......
......@@ -17,7 +17,7 @@ python test.py --cfg_file ${CONFIG_FILE} --batch_size ${BATCH_SIZE} --eval_all
* To test with multiple GPUs:
```shell script
sh scripts/slurm_test_mgpu.sh ${NUM_GPUS} \
sh scripts/slurm_test_mgpu.sh ${PARTITION} ${NUM_GPUS} \
--cfg_file ${CONFIG_FILE} --batch_size ${BATCH_SIZE}
```
......@@ -41,4 +41,4 @@ sh scripts/slurm_train.sh ${PARTITION} ${JOB_NAME} ${NUM_GPUS} \
* Train with a single GPU:
```shell script
python train.py --cfg_file ${CONFIG_FILE} --batch_size ${BATCH_SIZE} --epochs 50
```
\ No newline at end of file
```
......@@ -182,9 +182,7 @@ class KittiDataset(DatasetTemplate):
return info
# temp = process_single_scene(self.sample_id_list[0])
sample_id_list = sample_id_list if sample_id_list is not None else self.sample_id_list
sample_id_list = sample_id_list[:10]
with futures.ThreadPoolExecutor(num_workers) as executor:
infos = executor.map(process_single_scene, sample_id_list)
return list(infos)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment