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
d75836ea
Commit
d75836ea
authored
Dec 22, 2021
by
Tai-Wang
Browse files
Merge branch 'master' into v1.0.0.dev0
parents
022ee2fb
13f002d7
Changes
105
Hide whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
6 additions
and
17 deletions
+6
-17
setup.py
setup.py
+6
-1
tools/analysis_tools/get_flops.py
tools/analysis_tools/get_flops.py
+0
-4
tools/misc/browse_dataset.py
tools/misc/browse_dataset.py
+0
-4
tools/test.py
tools/test.py
+0
-4
tools/train.py
tools/train.py
+0
-4
No files found.
setup.py
View file @
d75836ea
from
setuptools
import
find_packages
,
setup
import
os
import
platform
import
shutil
import
sys
import
torch
...
...
@@ -150,7 +151,11 @@ def add_mim_extention():
# parse installment mode
if
'develop'
in
sys
.
argv
:
# installed by `pip install -e .`
mode
=
'symlink'
if
platform
.
system
()
==
'Windows'
:
# set `copy` mode here since symlink fails on Windows.
mode
=
'copy'
else
:
mode
=
'symlink'
elif
'sdist'
in
sys
.
argv
or
'bdist_wheel'
in
sys
.
argv
:
# installed by `pip install .`
# or create source distribution by `python setup.py sdist`
...
...
tools/analysis_tools/get_flops.py
View file @
d75836ea
...
...
@@ -62,10 +62,6 @@ def main():
cfg
=
Config
.
fromfile
(
args
.
config
)
if
args
.
cfg_options
is
not
None
:
cfg
.
merge_from_dict
(
args
.
cfg_options
)
# import modules from string list.
if
cfg
.
get
(
'custom_imports'
,
None
):
from
mmcv.utils
import
import_modules_from_strings
import_modules_from_strings
(
**
cfg
[
'custom_imports'
])
model
=
build_model
(
cfg
.
model
,
...
...
tools/misc/browse_dataset.py
View file @
d75836ea
...
...
@@ -55,10 +55,6 @@ def build_data_cfg(config_path, skip_type, cfg_options):
cfg
=
Config
.
fromfile
(
config_path
)
if
cfg_options
is
not
None
:
cfg
.
merge_from_dict
(
cfg_options
)
# import modules from string list.
if
cfg
.
get
(
'custom_imports'
,
None
):
from
mmcv.utils
import
import_modules_from_strings
import_modules_from_strings
(
**
cfg
[
'custom_imports'
])
# extract inner dataset of `RepeatDataset` as `cfg.data.train`
# so we don't need to worry about it later
if
cfg
.
data
.
train
[
'type'
]
==
'RepeatDataset'
:
...
...
tools/test.py
View file @
d75836ea
...
...
@@ -117,10 +117,6 @@ def main():
cfg
=
Config
.
fromfile
(
args
.
config
)
if
args
.
cfg_options
is
not
None
:
cfg
.
merge_from_dict
(
args
.
cfg_options
)
# import modules from string list.
if
cfg
.
get
(
'custom_imports'
,
None
):
from
mmcv.utils
import
import_modules_from_strings
import_modules_from_strings
(
**
cfg
[
'custom_imports'
])
# set cudnn_benchmark
if
cfg
.
get
(
'cudnn_benchmark'
,
False
):
torch
.
backends
.
cudnn
.
benchmark
=
True
...
...
tools/train.py
View file @
d75836ea
...
...
@@ -97,10 +97,6 @@ def main():
cfg
=
Config
.
fromfile
(
args
.
config
)
if
args
.
cfg_options
is
not
None
:
cfg
.
merge_from_dict
(
args
.
cfg_options
)
# import modules from string list.
if
cfg
.
get
(
'custom_imports'
,
None
):
from
mmcv.utils
import
import_modules_from_strings
import_modules_from_strings
(
**
cfg
[
'custom_imports'
])
# set cudnn_benchmark
if
cfg
.
get
(
'cudnn_benchmark'
,
False
):
...
...
Prev
1
2
3
4
5
6
Next
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