Skip to content

Commit

Permalink
[Fix]Fix dev-1.x branch ci (#1813)
Browse files Browse the repository at this point in the history
* add deploy.yaml

* fix ut

* fix update data script description
  • Loading branch information
VVsssssk authored Sep 16, 2022
1 parent 14ccfc4 commit aef0bfd
Show file tree
Hide file tree
Showing 6 changed files with 14 additions and 13 deletions.
File renamed without changes.
6 changes: 3 additions & 3 deletions tests/test_datasets/test_kitti_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,7 @@ def transform(self, info):
gt_instances_3d = InstanceData()
gt_instances_3d.labels_3d = info['gt_labels_3d']
data_sample.gt_instances_3d = gt_instances_3d
info['data_sample'] = data_sample
info['data_samples'] = data_sample
return info

pipeline = [
Expand Down Expand Up @@ -82,9 +82,9 @@ def test_getitem():
assert torch.allclose(ann_info['gt_bboxes_3d'].tensor.sum(),
torch.tensor(7.2650))
assert 'centers_2d' in ann_info
assert ann_info['centers_2d'].dtype == np.float64
assert ann_info['centers_2d'].dtype == np.float32
assert 'depths' in ann_info
assert ann_info['depths'].dtype == np.float64
assert ann_info['depths'].dtype == np.float32

car_kitti_dataset = KittiDataset(
data_root,
Expand Down
8 changes: 4 additions & 4 deletions tests/test_datasets/test_lyft_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,12 +21,12 @@ def _generate_nus_dataset_config():
class Identity(BaseTransform):

def transform(self, info):
packed_input = dict(data_sample=Det3DDataSample())
packed_input = dict(data_samples=Det3DDataSample())
if 'ann_info' in info:
packed_input['data_sample'].gt_instances_3d = InstanceData(
)
packed_input[
'data_sample'].gt_instances_3d.labels_3d = info[
'data_samples'].gt_instances_3d = InstanceData()
packed_input[
'data_samples'].gt_instances_3d.labels_3d = info[
'ann_info']['gt_labels_3d']
return packed_input

Expand Down
8 changes: 4 additions & 4 deletions tests/test_datasets/test_nuscenes_dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,12 +21,12 @@ def _generate_nus_dataset_config():
class Identity(BaseTransform):

def transform(self, info):
packed_input = dict(data_sample=Det3DDataSample())
packed_input = dict(data_samples=Det3DDataSample())
if 'ann_info' in info:
packed_input['data_sample'].gt_instances_3d = InstanceData(
)
packed_input[
'data_sample'].gt_instances_3d.labels_3d = info[
'data_samples'].gt_instances_3d = InstanceData()
packed_input[
'data_samples'].gt_instances_3d.labels_3d = info[
'ann_info']['gt_labels_3d']
return packed_input

Expand Down
4 changes: 2 additions & 2 deletions tests/test_models/test_dense_heads/test_freeanchors.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,8 @@ def test_freeanchor(self):
DefaultScope.get_instance('test_freeanchor', scope_name='mmdet3d')
_setup_seed(0)
freeanchor_cfg = _get_detector_cfg(
'free_anchor/hv_pointpillars_fpn_sbn-all_free-'
'anchor_4x8_2x_nus-3d.py')
'free_anchor/pointpillars_hv_regnet-1.6gf_fpn_head-free-anchor'
'_sbn-all_8xb4-2x_nus-3d.py')
model = MODELS.build(freeanchor_cfg)
num_gt_instance = 3
packed_inputs = _create_detector_inputs(
Expand Down
1 change: 1 addition & 0 deletions tools/dataset_converters/update_infos_to_v2.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@
Example:
python tools/dataset_converters/update_infos_to_v2.py
--dataset kitti
--pkl ./data/kitti/kitti_infos_train.pkl
--out-dir ./kitti_v2/
"""
Expand Down

0 comments on commit aef0bfd

Please sign in to comment.