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