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 ...@@ -7,7 +7,7 @@ As of now, it mainly consists of `PCDet` toolbox for 3D object detection from po
### What does `PCDet` toolbox do? ### 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. `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. 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 ...@@ -17,7 +17,7 @@ python test.py --cfg_file ${CONFIG_FILE} --batch_size ${BATCH_SIZE} --eval_all
* To test with multiple GPUs: * To test with multiple GPUs:
```shell script ```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} --cfg_file ${CONFIG_FILE} --batch_size ${BATCH_SIZE}
``` ```
...@@ -41,4 +41,4 @@ sh scripts/slurm_train.sh ${PARTITION} ${JOB_NAME} ${NUM_GPUS} \ ...@@ -41,4 +41,4 @@ sh scripts/slurm_train.sh ${PARTITION} ${JOB_NAME} ${NUM_GPUS} \
* Train with a single GPU: * Train with a single GPU:
```shell script ```shell script
python train.py --cfg_file ${CONFIG_FILE} --batch_size ${BATCH_SIZE} --epochs 50 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): ...@@ -182,9 +182,7 @@ class KittiDataset(DatasetTemplate):
return info 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 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: with futures.ThreadPoolExecutor(num_workers) as executor:
infos = executor.map(process_single_scene, sample_id_list) infos = executor.map(process_single_scene, sample_id_list)
return list(infos) 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