Commit a225e961 authored by zhangwenwei's avatar zhangwenwei
Browse files

Fix tweaks

parent 0d385553
...@@ -63,7 +63,7 @@ model = dict( ...@@ -63,7 +63,7 @@ model = dict(
type='SECONDFPN', type='SECONDFPN',
in_channels=[128, 256], in_channels=[128, 256],
upsample_strides=[1, 2], upsample_strides=[1, 2],
num_upsample_filters=[256, 256], out_channels=[256, 256],
), ),
pts_bbox_head=dict( pts_bbox_head=dict(
type='SECONDHead', type='SECONDHead',
......
...@@ -41,7 +41,7 @@ model = dict( ...@@ -41,7 +41,7 @@ model = dict(
norm_cfg=dict(type='naiveSyncBN2d', eps=1e-3, momentum=0.01), norm_cfg=dict(type='naiveSyncBN2d', eps=1e-3, momentum=0.01),
in_channels=[64, 128, 256], in_channels=[64, 128, 256],
upsample_strides=[1, 2, 4], upsample_strides=[1, 2, 4],
num_upsample_filters=[128, 128, 128], out_channels=[128, 128, 128],
), ),
pts_bbox_head=dict( pts_bbox_head=dict(
type='Anchor3DVeloHead', type='Anchor3DVeloHead',
......
...@@ -8,7 +8,7 @@ def test_secfpn(): ...@@ -8,7 +8,7 @@ def test_secfpn():
upsample_strides=[1, 2], upsample_strides=[1, 2],
out_channels=[4, 6], out_channels=[4, 6],
) )
from mmdet.models.builders import build_neck from mmdet.models.builder import build_neck
neck = build_neck(neck_cfg) neck = build_neck(neck_cfg)
assert neck.deblocks[0][0].in_channels == 2 assert neck.deblocks[0][0].in_channels == 2
assert neck.deblocks[1][0].in_channels == 3 assert neck.deblocks[1][0].in_channels == 3
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment