2 Star 5 Fork 1

OpenMMLab / mmfewshot

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
meta-rcnn_r101_c4_8xb4_voc-split1_5shot-fine-tuning.py 1.21 KB
一键复制 编辑 原始数据 按行查看 历史
Linyiqi 提交于 2021-11-22 14:57 . fix typos & update README (#55)
_base_ = [
'../../../_base_/datasets/nway_kshot/few_shot_voc.py',
'../../../_base_/schedules/schedule.py', '../../meta-rcnn_r101_c4.py',
'../../../_base_/default_runtime.py'
]
# classes splits are predefined in FewShotVOCDataset
# FewShotVOCDefaultDataset predefine ann_cfg for model reproducibility.
data = dict(
train=dict(
save_dataset=True,
dataset=dict(
type='FewShotVOCDefaultDataset',
ann_cfg=[dict(method='MetaRCNN', setting='SPLIT1_5SHOT')],
num_novel_shots=5,
num_base_shots=5,
classes='ALL_CLASSES_SPLIT1',
)),
val=dict(classes='ALL_CLASSES_SPLIT1'),
test=dict(classes='ALL_CLASSES_SPLIT1'),
model_init=dict(classes='ALL_CLASSES_SPLIT1'))
evaluation = dict(
interval=200, class_splits=['BASE_CLASSES_SPLIT1', 'NOVEL_CLASSES_SPLIT1'])
checkpoint_config = dict(interval=200)
optimizer = dict(lr=0.001)
lr_config = dict(warmup=None)
runner = dict(max_iters=800)
# load_from = 'path of base training model'
load_from = \
'work_dirs/meta-rcnn_r101_c4_8xb4_voc-split1_base-training/latest.pth'
# model settings
model = dict(frozen_parameters=[
'backbone', 'shared_head', 'rpn_head', 'aggregation_layer'
])
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/open-mmlab/mmfewshot.git
git@gitee.com:open-mmlab/mmfewshot.git
open-mmlab
mmfewshot
mmfewshot
main

搜索帮助

344bd9b3 5694891 D2dac590 5694891