2 Star 5 Fork 1

OpenMMLab / mmfewshot

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
meta-rcnn_r101_c4_8xb4_voc-split3_base-training.py 907 Bytes
一键复制 编辑 原始数据 按行查看 历史
Linyiqi 提交于 2021-11-22 14:57 . fix typos & update README (#55)
_base_ = [
'../../../_base_/datasets/nway_kshot/base_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=False,
dataset=dict(classes='BASE_CLASSES_SPLIT3'),
support_dataset=dict(classes='BASE_CLASSES_SPLIT3')),
val=dict(classes='BASE_CLASSES_SPLIT3'),
test=dict(classes='BASE_CLASSES_SPLIT3'),
model_init=dict(classes='BASE_CLASSES_SPLIT3'))
lr_config = dict(warmup_iters=100, step=[16000])
evaluation = dict(interval=6000)
checkpoint_config = dict(interval=6000)
runner = dict(max_iters=18000)
optimizer = dict(lr=0.005)
# model settings
model = dict(
roi_head=dict(bbox_head=dict(num_classes=15, num_meta_classes=15)))
马建仓 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