Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
I
ICV-mmengine_basecode
Manage
Activity
Members
Labels
Plan
Issues
Issue boards
Milestones
Wiki
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Snippets
Build
Pipelines
Jobs
Pipeline schedules
Artifacts
Deploy
Releases
Package Registry
Container Registry
Model registry
Operate
Environments
Terraform modules
Monitor
Incidents
Service Desk
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
Florian Schiffel
ICV-mmengine_basecode
Commits
817eb89a
Unverified
Commit
817eb89a
authored
3 years ago
by
Wenwei Zhang
Committed by
GitHub
3 years ago
Browse files
Options
Downloads
Patches
Plain Diff
Add user docs for get_config/get_models (#83)
* add user docs for get_config/get_models * resolve comments
parent
fd851564
No related branches found
No related tags found
No related merge requests found
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
docs/zh_cn/tutorials/config.md
+52
-0
52 additions, 0 deletions
docs/zh_cn/tutorials/config.md
with
52 additions
and
0 deletions
docs/zh_cn/tutorials/config.md
+
52
−
0
View file @
817eb89a
...
@@ -222,6 +222,58 @@ cfg.work_dir # "./work_dir/config_setting1"
...
@@ -222,6 +222,58 @@ cfg.work_dir # "./work_dir/config_setting1"
-
`{{ fileBasenameNoExtension }}`
- 当前文件不包含扩展名的文件名,例如 file
-
`{{ fileBasenameNoExtension }}`
- 当前文件不包含扩展名的文件名,例如 file
-
`{{ fileExtname }}`
- 当前文件的扩展名,例如
`.py`
-
`{{ fileExtname }}`
- 当前文件的扩展名,例如
`.py`
### 跨项目继承配置文件
为了避免基于已有算法库开发的新项目复制大量的配置文件,MMEngine 中的 config 支持配置文件的跨项目继承。
假定 MMDetection 项目中存在如下配置文件
```
text
configs/_base_/schedules/schedule_1x.py
configs/_base_/datasets.coco_instance.py
configs/_base_/default_runtime.py
configs/_base_/models/faster_rcnn_r50_fpn.py
```
在 MMDetection 被安装进环境(如使用
`pip install mmdet`
)以后,新的项目可以直接在自己的配置文件中继承 MMDetection 的配置文件而无需拷贝,使用方式如下所示
```
python
_base_
=
[
'
mmdet::_base_/schedules/schedule_1x.py
'
,
'
mmdet::_base_/datasets.coco_instance.py
'
,
'
mmdet::_base_/default_runtime.py
'
'
mmdet::_base_/models/faster_rcnn_r50_fpn.py
'
,
]
```
通过指定
`mmdet::`
,Config 类会去检索 mmdet 包中的配置文件目录,并继承指定的配置文件。
实际上,只要算法库的
`setup.py`
文件符合
[
MMEngine 安装规范
](
todo
)
,在正确安装算法库以后,新的项目就可以使用上述用法去继承已有算法库的配置文件而无需拷贝。
### 跨项目使用配置文件
MMEngine 还提供了
`get_config`
和
`get_model`
两个接口,支持对符合
[
MMEngine 安装规范
](
todo
)
的算法库中的模型和配置文件做索引并进行 API 调用。通过
`get_model`
接口可以获得构建好的模型。通过
`get_config`
接口可以获得配置文件。
`get_model`
的使用样例如下所示,使用和跨项目继承配置文件相同的语法,指定
`mmdet::`
,即可在 mmdet 包中检索对应的配置文件并构建和初始化相应模型。
用户可以通过指定
`pretrained=True`
获得已经加载预训练权重的模型以进行训练或者推理。
```
python
from
mmengine
import
get_model
model
=
get_model
(
'
mmdet::faster_rcnn/faster_rcnn_r50_fpn_1x_coco
'
,
pretrained
=
True
)
```
`get_config`
的使用样例如下所示,使用和跨项目继承配置文件相同的语法,指定
`mmdet::`
,即可实现去 mmdet 包中检索并加载对应的配置文件。
用户可以基于这样得到的配置文件进行推理修改并自定义自己的算法模型。
同时,如果用户指定
`pretrained=True`
,得到的配置文件中会新增
`model_path`
字段,指定了对应模型预训练权重的路径。
```
python
from
mmengine
import
get_config
cfg
=
get_config
(
'
mmdet::faster_rcnn/faster_rcnn_r50_fpn_1x_coco
'
,
pretrained
=
True
)
model_path
=
cfg
.
model_path
from
mmdet.models
import
build_model
model
=
build_model
(
cfg
.
model
)
load_checkpoint
(
model
,
model_path
)
```
### 导入自定义 Python 模块
### 导入自定义 Python 模块
将配置与注册器结合起来使用时,如果我们往注册器中注册了一些自定义的类,就可能会遇到一些问题。因为读取配置文件的时候,这部分代码可能还没有被执行到,所以并未完成注册过程,从而导致构建自定义类的时候报错。
将配置与注册器结合起来使用时,如果我们往注册器中注册了一些自定义的类,就可能会遇到一些问题。因为读取配置文件的时候,这部分代码可能还没有被执行到,所以并未完成注册过程,从而导致构建自定义类的时候报错。
...
...
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment