Skip to content

GitLab

Commit 0868596e authored by Kai Chen's avatar Kai Chen
Browse files

bug fix for missing arguments

parent c8cc01e8
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
mmdet/models/bbox_heads/convfc_bbox_head.py
View file @ 0868596e
...@@ -22,6 +22,7 @@ class ConvFCBBoxHead(BBoxHead): ...@@ -22,6 +22,7 @@ class ConvFCBBoxHead(BBoxHead):
num_reg_fcs=0, num_reg_fcs=0,
conv_out_channels=256, conv_out_channels=256,
fc_out_channels=1024, fc_out_channels=1024,
normalize=None,
*args, *args,
**kwargs): **kwargs):
super(ConvFCBBoxHead, self).__init__(*args, **kwargs) super(ConvFCBBoxHead, self).__init__(*args, **kwargs)
...@@ -41,6 +42,8 @@ class ConvFCBBoxHead(BBoxHead): ...@@ -41,6 +42,8 @@ class ConvFCBBoxHead(BBoxHead):
self.num_reg_fcs = num_reg_fcs self.num_reg_fcs = num_reg_fcs
self.conv_out_channels = conv_out_channels self.conv_out_channels = conv_out_channels
self.fc_out_channels = fc_out_channels self.fc_out_channels = fc_out_channels
self.normalize = normalize
self.with_bias = normalize is None
# add shared convs and fcs # add shared convs and fcs
self.shared_convs, self.shared_fcs, last_layer_dim = \ self.shared_convs, self.shared_fcs, last_layer_dim = \
......
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