"vscode:/vscode.git/clone" did not exist on "ca11d10806f024f9c8ae8e397ac7a429c4e60041"
Unverified Commit 5592a505 authored by Yezhen Cong's avatar Yezhen Cong Committed by GitHub
Browse files

[Fix] Fix typos and error in comments (#308)

* fix bug caused by mmcv upgrade; delete pdb breakpoint

* fix typos
parent ac9a3e8c
...@@ -52,7 +52,7 @@ model = dict( ...@@ -52,7 +52,7 @@ model = dict(
out_size_factor=8, out_size_factor=8,
voxel_size=voxel_size[:2], voxel_size=voxel_size[:2],
code_size=9), code_size=9),
seperate_head=dict( separate_head=dict(
type='SeparateHead', init_bias=-2.19, final_kernel=3), type='SeparateHead', init_bias=-2.19, final_kernel=3),
loss_cls=dict(type='GaussianFocalLoss', reduction='mean'), loss_cls=dict(type='GaussianFocalLoss', reduction='mean'),
loss_bbox=dict(type='L1Loss', reduction='mean', loss_weight=0.25), loss_bbox=dict(type='L1Loss', reduction='mean', loss_weight=0.25),
......
...@@ -51,7 +51,7 @@ model = dict( ...@@ -51,7 +51,7 @@ model = dict(
out_size_factor=4, out_size_factor=4,
voxel_size=voxel_size[:2], voxel_size=voxel_size[:2],
code_size=9), code_size=9),
seperate_head=dict( separate_head=dict(
type='SeparateHead', init_bias=-2.19, final_kernel=3), type='SeparateHead', init_bias=-2.19, final_kernel=3),
loss_cls=dict(type='GaussianFocalLoss', reduction='mean'), loss_cls=dict(type='GaussianFocalLoss', reduction='mean'),
loss_bbox=dict(type='L1Loss', reduction='mean', loss_weight=0.25), loss_bbox=dict(type='L1Loss', reduction='mean', loss_weight=0.25),
......
...@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_0075voxel_second_secfpn_4x8_cyclic_20e_nus.py'] ...@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_0075voxel_second_secfpn_4x8_cyclic_20e_nus.py']
model = dict( model = dict(
pts_bbox_head=dict( pts_bbox_head=dict(
seperate_head=dict( separate_head=dict(
type='DCNSeperateHead', type='DCNSeparateHead',
dcn_config=dict( dcn_config=dict(
type='DCN', type='DCN',
in_channels=64, in_channels=64,
......
...@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_0075voxel_second_secfpn_4x8_cyclic_20e_nus.py'] ...@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_0075voxel_second_secfpn_4x8_cyclic_20e_nus.py']
model = dict( model = dict(
pts_bbox_head=dict( pts_bbox_head=dict(
seperate_head=dict( separate_head=dict(
type='DCNSeperateHead', type='DCNSeparateHead',
dcn_config=dict( dcn_config=dict(
type='DCN', type='DCN',
in_channels=64, in_channels=64,
......
...@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_01voxel_second_secfpn_4x8_cyclic_20e_nus.py'] ...@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_01voxel_second_secfpn_4x8_cyclic_20e_nus.py']
model = dict( model = dict(
pts_bbox_head=dict( pts_bbox_head=dict(
seperate_head=dict( separate_head=dict(
type='DCNSeperateHead', type='DCNSeparateHead',
dcn_config=dict( dcn_config=dict(
type='DCN', type='DCN',
in_channels=64, in_channels=64,
......
...@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_01voxel_second_secfpn_4x8_cyclic_20e_nus.py'] ...@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_01voxel_second_secfpn_4x8_cyclic_20e_nus.py']
model = dict( model = dict(
pts_bbox_head=dict( pts_bbox_head=dict(
seperate_head=dict( separate_head=dict(
type='DCNSeperateHead', type='DCNSeparateHead',
dcn_config=dict( dcn_config=dict(
type='DCN', type='DCN',
in_channels=64, in_channels=64,
......
...@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_02pillar_second_secfpn_4x8_cyclic_20e_nus.py'] ...@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_02pillar_second_secfpn_4x8_cyclic_20e_nus.py']
model = dict( model = dict(
pts_bbox_head=dict( pts_bbox_head=dict(
seperate_head=dict( separate_head=dict(
type='DCNSeperateHead', type='DCNSeparateHead',
dcn_config=dict( dcn_config=dict(
type='DCN', type='DCN',
in_channels=64, in_channels=64,
......
...@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_02pillar_second_secfpn_4x8_cyclic_20e_nus.py'] ...@@ -2,8 +2,8 @@ _base_ = ['./centerpoint_02pillar_second_secfpn_4x8_cyclic_20e_nus.py']
model = dict( model = dict(
pts_bbox_head=dict( pts_bbox_head=dict(
seperate_head=dict( separate_head=dict(
type='DCNSeperateHead', type='DCNSeparateHead',
dcn_config=dict( dcn_config=dict(
type='DCN', type='DCN',
in_channels=64, in_channels=64,
......
...@@ -119,8 +119,8 @@ class SeparateHead(nn.Module): ...@@ -119,8 +119,8 @@ class SeparateHead(nn.Module):
@HEADS.register_module() @HEADS.register_module()
class DCNSeperateHead(nn.Module): class DCNSeparateHead(nn.Module):
r"""DCNSeperateHead for CenterHead. r"""DCNSeparateHead for CenterHead.
.. code-block:: none .. code-block:: none
/-----> DCN for heatmap task -----> heatmap task. /-----> DCN for heatmap task -----> heatmap task.
...@@ -155,7 +155,7 @@ class DCNSeperateHead(nn.Module): ...@@ -155,7 +155,7 @@ class DCNSeperateHead(nn.Module):
norm_cfg=dict(type='BN2d'), norm_cfg=dict(type='BN2d'),
bias='auto', bias='auto',
**kwargs): **kwargs):
super(DCNSeperateHead, self).__init__() super(DCNSeparateHead, self).__init__()
if 'heatmap' in heads: if 'heatmap' in heads:
heads.pop('heatmap') heads.pop('heatmap')
# feature adaptation with dcn # feature adaptation with dcn
...@@ -250,7 +250,7 @@ class CenterHead(nn.Module): ...@@ -250,7 +250,7 @@ class CenterHead(nn.Module):
Default: dict(type='GaussianFocalLoss', reduction='mean'). Default: dict(type='GaussianFocalLoss', reduction='mean').
loss_bbox (dict): Config of regression loss function. loss_bbox (dict): Config of regression loss function.
Default: dict(type='L1Loss', reduction='none'). Default: dict(type='L1Loss', reduction='none').
seperate_head (dict): Config of seperate head. Default: dict( separate_head (dict): Config of separate head. Default: dict(
type='SeparateHead', init_bias=-2.19, final_kernel=3) type='SeparateHead', init_bias=-2.19, final_kernel=3)
share_conv_channel (int): Output channels for share_conv_layer. share_conv_channel (int): Output channels for share_conv_layer.
Default: 64. Default: 64.
...@@ -273,7 +273,7 @@ class CenterHead(nn.Module): ...@@ -273,7 +273,7 @@ class CenterHead(nn.Module):
loss_cls=dict(type='GaussianFocalLoss', reduction='mean'), loss_cls=dict(type='GaussianFocalLoss', reduction='mean'),
loss_bbox=dict( loss_bbox=dict(
type='L1Loss', reduction='none', loss_weight=0.25), type='L1Loss', reduction='none', loss_weight=0.25),
seperate_head=dict( separate_head=dict(
type='SeparateHead', init_bias=-2.19, final_kernel=3), type='SeparateHead', init_bias=-2.19, final_kernel=3),
share_conv_channel=64, share_conv_channel=64,
num_heatmap_convs=2, num_heatmap_convs=2,
...@@ -312,9 +312,9 @@ class CenterHead(nn.Module): ...@@ -312,9 +312,9 @@ class CenterHead(nn.Module):
for num_cls in num_classes: for num_cls in num_classes:
heads = copy.deepcopy(common_heads) heads = copy.deepcopy(common_heads)
heads.update(dict(heatmap=(num_cls, num_heatmap_convs))) heads.update(dict(heatmap=(num_cls, num_heatmap_convs)))
seperate_head.update( separate_head.update(
in_channels=share_conv_channel, heads=heads, num_cls=num_cls) in_channels=share_conv_channel, heads=heads, num_cls=num_cls)
self.task_heads.append(builder.build_head(seperate_head)) self.task_heads.append(builder.build_head(separate_head))
def init_weights(self): def init_weights(self):
"""Initialize weights.""" """Initialize weights."""
......
...@@ -291,7 +291,7 @@ class PointFusion(nn.Module): ...@@ -291,7 +291,7 @@ class PointFusion(nn.Module):
Args: Args:
img_feats (torch.Tensor): Image feature map in shape img_feats (torch.Tensor): Image feature map in shape
(N, C, H, W). (1, C, H, W).
pts (torch.Tensor): Points of a single sample. pts (torch.Tensor): Points of a single sample.
img_meta (dict): Meta information of the single sample. img_meta (dict): Meta information of the single sample.
......
"""Test model forward process. """Test model forward process.
CommandLine: CommandLine:
pytest tests/test_forward.py pytest tests/test_models/test_forward.py
xdoctest tests/test_forward.py zero xdoctest tests/test_models/test_forward.py zero
""" """
import copy import copy
import numpy as np import numpy as np
......
...@@ -806,8 +806,8 @@ def test_dcn_center_head(): ...@@ -806,8 +806,8 @@ def test_dcn_center_head():
out_size_factor=4, out_size_factor=4,
voxel_size=voxel_size[:2], voxel_size=voxel_size[:2],
code_size=9), code_size=9),
seperate_head=dict( separate_head=dict(
type='DCNSeperateHead', type='DCNSeparateHead',
dcn_config=dict( dcn_config=dict(
type='DCN', type='DCN',
in_channels=64, in_channels=64,
......
""" """
CommandLine: CommandLine:
pytest tests/test_anchor.py pytest tests/test_utils/test_anchor.py
xdoctest tests/test_anchor.py zero xdoctest tests/test_utils/test_anchor.py zero
""" """
import torch import torch
......
"""Tests the Assigner objects. """Tests the Assigner objects.
CommandLine: CommandLine:
pytest tests/test_assigner.py pytest tests/test_utils/test_assigner.py
xdoctest tests/test_assigner.py zero xdoctest tests/test_utils/test_assigner.py zero
""" """
import torch import torch
......
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