Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

关于resnet101的config文件配置 #77

Open
Amireux52 opened this issue Nov 26, 2024 · 2 comments
Open

关于resnet101的config文件配置 #77

Amireux52 opened this issue Nov 26, 2024 · 2 comments

Comments

@Amireux52
Copy link

作者您好,感谢您工作的贡献。如果我要用resnet101作为主干,请问我修改配置文件内容如下可以吗:
img_backbone=dict(
type="ResNet",
depth=101,
num_stages=4,
out_indices=(1, 2, 3),
frozen_stages=4,
norm_cfg=dict(type="BN2d", requires_grad=False),
norm_eval=True,
style="caffe",
dcn=dict(
type="DCNv2", deform_groups=1, fallback_on_stride=False
), # original DCNv2 will print log when perform load_state_dict
stage_with_dcn=(False, False, True, True),
),

@swc-17
Copy link
Owner

swc-17 commented Nov 28, 2024

 img_backbone=dict(
        type='ResNet',
        depth=101,
        num_stages=4,
        frozen_stages=-1,
        style='pytorch',
        with_cp=True,
        out_indices=(0, 1, 2, 3),
        norm_eval=True,
        norm_cfg=dict(type='BN', requires_grad=False),
        init_cfg=dict(
            type='Pretrained',
            checkpoint=
            'ckpt/cascade_mask_rcnn_r101_fpn_1x_nuim_20201024_134804-45215b1e.pth',
            prefix='backbone.')),
    img_neck=dict(
        type='FPN',
        num_outs=4,
        start_level=1,
        out_channels=256,
        add_extra_convs='on_output',
        relu_before_extra_convs=True,
        in_channels=[256, 512, 1024, 2048]
    ),

@Amireux52
Copy link
Author

好嘞,我试试,谢谢啦

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants