cfg, default_args=dict(train_cfg=train_cfg, test_cfg=test_cfg)) DETECTORS实际上是MODELS,继续查到基础的mmcv库,MODELS实际上就是Registry类的一个实例,Registry类的build方法实际调用的是build_from_cfg函数,核心代码如下。 import inspect def build_from_cfg(cfg, registry, default_args=None): # 删去了...
train_pipeline=[dict(type='LoadImageFromFile'),dict(type='LoadAnnotations',with_bbox=True),dict(type='Resize',img_scale=(1333,800),keep_ratio=True),dict(type='RandomFlip',flip_ratio=0.5),dict(type='Normalize',**img_norm_cfg),dict(type='Pad',size_divisor=32),dict(type='MyTransform'...
下面是配置文件test.py的示例。 a = 1 b = dict(b1=[0, 1, 2], b2=None) c = (1, 2) d = 'string' >>> cfg = Config.fromfile('test.py') >>> print(cfg) >>> dict(a=1, ... b=dict(b1=[0, 1, 2], b2=None), ... c=(1, 2), ... d='string') 对于所有的配置格...
首先进入Runer类中的test函数: 然后进入Runer类中的build_test_loop函数: 然后经过Registry中的build_from_cfg等函数,进入TestLoop类的__init__进行初始化。初始化的时候,会进入父类BaseLoop,在BaseLoop中,会对Dataloader进行build,关于mmcv2.0是如何构建Dataloader的,可以参考:mmcv2.0中build dataloader的流程 - Picass...
- setup.cfg定义的代码静态检查时使用的规则; - .github目录定义了 github 提交的 workflow 和 Issue/PR 模板。 2. 算法库设计小技巧 新的算法库开始设计时,应当着重定义两大模块 dataset 和 model 的抽象接口,如果是基于已有算法库的拓展,一般项目都会保持和依赖算法库一样的输入输出流并尽可能复用已有模块,避...
Breadcrumbs mmcv / setup.cfg Latest commit zhouzaida Upgrade the versions of pre-commit hooks (open-mmlab#2328) 75ea2f8· Oct 11, 2022 HistoryHistory File metadata and controls Code Blame 26 lines (22 loc) · 755 Bytes Raw [bdist_wheel] universal=1 [aliases] test=pytest [yapf] based_...
model, test_cfg=config.test_cfg) if checkpoint is not None: checkpoint = load_checkpoint(model, checkpoint) if 'CLASSES' in checkpoint['meta']: model.CLASSES = checkpoint['meta']['CLASSES'] else: warnings.warn('Class names are not saved in the checkpoint\'s ' 'meta data, use COCO ...
# test mmcv with both CPU and CUDA ops if torch.cuda.is_available(): boxes1 = boxes1.cuda() boxes2 = boxes2.cuda() @@ -33,9 +33,9 @@ def check_installation(): if __name__ == '__main__': print('Start checking the installation of mmcv-full ...') print('Start checking...
- setup.cfg定义的代码静态检查时使用的规则; - .github目录定义了 github 提交的 workflow 和 Issue/PR 模板。 2. 算法库设计小技巧 新的算法库开始设计时,应当着重定义两大模块 dataset 和 model 的抽象接口,如果是基于已有算法库的拓展,一般项目都会保持和依赖算法库一样的输入输出流并尽可能复用已有模块,避...
test: type:folder path:test/ optimizer: type:sgd lr:0.01 weight_decay:0.0001 配置文件中的每个字段都具有自己的作用。其中,`model`字段定义了模型相关的配置,`dataset`字段定义了数据集相关的配置,`optimizer`字段定义了优化器相关的配置。用户可以根据具体任务进行配置的修改。 3. 参数设置 在配置文件中,可以...