Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
OpenPCDet
Commits
a35f4297
Unverified
Commit
a35f4297
authored
Mar 12, 2022
by
jihan.yang
Committed by
GitHub
Mar 12, 2022
Browse files
slight modification for lyft (#863)
parent
9c8e583e
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
13 additions
and
4 deletions
+13
-4
docs/GETTING_STARTED.md
docs/GETTING_STARTED.md
+9
-2
pcdet/datasets/lyft/lyft_dataset.py
pcdet/datasets/lyft/lyft_dataset.py
+4
-2
No files found.
docs/GETTING_STARTED.md
View file @
a35f4297
...
@@ -108,7 +108,9 @@ OpenPCDet
...
@@ -108,7 +108,9 @@ OpenPCDet
│ ├── lyft
│ ├── lyft
│ │ │── ImageSets
│ │ │── ImageSets
│ │ │── trainval
│ │ │── trainval
│ │ │ │── data & maps & images & lidar & train_lidar
│ │ │ │── data & maps(train_maps) & images(train_images) & lidar(train_lidar) & train_lidar
│ │ │── test
│ │ │ │── data & maps(test_maps) & test_images & test_lidar
├── pcdet
├── pcdet
├── tools
├── tools
```
```
...
@@ -118,11 +120,16 @@ OpenPCDet
...
@@ -118,11 +120,16 @@ OpenPCDet
pip install -U lyft_dataset_sdk==0.0.8
pip install -U lyft_dataset_sdk==0.0.8
```
```
*
Generate the data infos by running the following command (it may take several hours):
*
Generate the
training & validation
data infos by running the following command (it may take several hours):
```
python
```
python
python
-
m
pcdet
.
datasets
.
lyft
.
lyft_dataset
--
func
create_lyft_infos
\
python
-
m
pcdet
.
datasets
.
lyft
.
lyft_dataset
--
func
create_lyft_infos
\
--
cfg_file
tools
/
cfgs
/
dataset_configs
/
lyft_dataset
.
yaml
--
cfg_file
tools
/
cfgs
/
dataset_configs
/
lyft_dataset
.
yaml
```
```
*
Generate the test data infos by running the following command:
```
python
python
-
m
pcdet
.
datasets
.
lyft
.
lyft_dataset
--
func
create_lyft_infos
\
--
cfg_file
tools
/
cfgs
/
dataset_configs
/
lyft_dataset
.
yaml
--
version
test
```
*
You need to check carefully since we don't provide a benchmark for it.
*
You need to check carefully since we don't provide a benchmark for it.
...
...
pcdet/datasets/lyft/lyft_dataset.py
View file @
a35f4297
...
@@ -276,7 +276,7 @@ def create_lyft_info(version, data_path, save_path, split, max_sweeps=10):
...
@@ -276,7 +276,7 @@ def create_lyft_info(version, data_path, save_path, split, max_sweeps=10):
raise
NotImplementedError
raise
NotImplementedError
train_scenes
=
[
x
.
strip
()
for
x
in
open
(
train_split_path
).
readlines
()]
if
train_split_path
.
exists
()
else
[]
train_scenes
=
[
x
.
strip
()
for
x
in
open
(
train_split_path
).
readlines
()]
if
train_split_path
.
exists
()
else
[]
val_scenes
=
[
x
.
strip
()
for
x
in
open
(
val_split_path
).
readlines
()]
if
val_split_path
.
exists
()
else
[]
val_scenes
=
[
x
.
strip
()
for
x
in
open
(
val_split_path
).
readlines
()]
if
val_split_path
is
not
None
and
val_split_path
.
exists
()
else
[]
lyft
=
LyftDataset
(
json_path
=
data_path
/
'data'
,
data_path
=
data_path
,
verbose
=
True
)
lyft
=
LyftDataset
(
json_path
=
data_path
/
'data'
,
data_path
=
data_path
,
verbose
=
True
)
...
@@ -342,4 +342,6 @@ if __name__ == '__main__':
...
@@ -342,4 +342,6 @@ if __name__ == '__main__':
root_path
=
ROOT_DIR
/
'data'
/
'lyft'
,
root_path
=
ROOT_DIR
/
'data'
/
'lyft'
,
logger
=
common_utils
.
create_logger
(),
training
=
True
logger
=
common_utils
.
create_logger
(),
training
=
True
)
)
lyft_dataset
.
create_groundtruth_database
(
max_sweeps
=
dataset_cfg
.
MAX_SWEEPS
)
if
args
.
version
!=
'test'
:
lyft_dataset
.
create_groundtruth_database
(
max_sweeps
=
dataset_cfg
.
MAX_SWEEPS
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment