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
2492836b
Unverified
Commit
2492836b
authored
Oct 10, 2020
by
yinchimaoliang
Committed by
GitHub
Oct 10, 2020
Browse files
[Fix]: gather-models (#153)
* Change gather_models script. * Add usage.
parent
86fa43ba
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
8 additions
and
1 deletion
+8
-1
.dev_scripts/gather_models.py
.dev_scripts/gather_models.py
+8
-1
No files found.
.dev_scripts/gather_models.py
View file @
2492836b
"""Script to gather benchmarked models and prepare them for upload.
Usage:
python gather_models.py ${root_path} ${out_dir}
"""
import
argparse
import
glob
import
json
...
...
@@ -163,7 +169,8 @@ def main():
model_publish_dir
=
osp
.
join
(
models_out
,
model
[
'config'
].
rstrip
(
'.py'
))
mmcv
.
mkdir_or_exist
(
model_publish_dir
)
model_name
=
model_publish_dir
+
'_'
+
model
[
'model_time'
]
model_name
=
model
[
'config'
].
split
(
'/'
)[
-
1
].
rstrip
(
'.py'
)
+
'_'
+
model
[
'model_time'
]
publish_model_path
=
osp
.
join
(
model_publish_dir
,
model_name
)
trained_model_path
=
osp
.
join
(
models_root
,
model
[
'config'
],
'epoch_{}.pth'
.
format
(
model
[
'epochs'
]))
...
...
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