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
mmdetection3d
Commits
1c72acdd
Unverified
Commit
1c72acdd
authored
Jun 08, 2022
by
ChaimZhu
Committed by
GitHub
Jun 08, 2022
Browse files
[Fix]: fix coco registry problem (#1536)
parent
fab3db2e
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
2 deletions
+4
-2
mmdet3d/datasets/builder.py
mmdet3d/datasets/builder.py
+4
-2
No files found.
mmdet3d/datasets/builder.py
View file @
1c72acdd
...
@@ -3,6 +3,7 @@ import platform
...
@@ -3,6 +3,7 @@ import platform
from
mmcv.utils
import
Registry
,
build_from_cfg
from
mmcv.utils
import
Registry
,
build_from_cfg
from
mmdet.datasets
import
DATASETS
as
MMDET_DATASETS
from
mmdet.datasets.builder
import
_concat_dataset
from
mmdet.datasets.builder
import
_concat_dataset
if
platform
.
system
()
!=
'Windows'
:
if
platform
.
system
()
!=
'Windows'
:
...
@@ -39,7 +40,8 @@ def build_dataset(cfg, default_args=None):
...
@@ -39,7 +40,8 @@ def build_dataset(cfg, default_args=None):
dataset
=
CBGSDataset
(
build_dataset
(
cfg
[
'dataset'
],
default_args
))
dataset
=
CBGSDataset
(
build_dataset
(
cfg
[
'dataset'
],
default_args
))
elif
isinstance
(
cfg
.
get
(
'ann_file'
),
(
list
,
tuple
)):
elif
isinstance
(
cfg
.
get
(
'ann_file'
),
(
list
,
tuple
)):
dataset
=
_concat_dataset
(
cfg
,
default_args
)
dataset
=
_concat_dataset
(
cfg
,
default_args
)
el
se
:
el
if
cfg
[
'type'
]
in
DATASETS
.
_module_dict
.
keys
()
:
dataset
=
build_from_cfg
(
cfg
,
DATASETS
,
default_args
)
dataset
=
build_from_cfg
(
cfg
,
DATASETS
,
default_args
)
else
:
dataset
=
build_from_cfg
(
cfg
,
MMDET_DATASETS
,
default_args
)
return
dataset
return
dataset
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