Unverified Commit 583c4acc authored by Qing Lian's avatar Qing Lian Committed by GitHub
Browse files

Update waymo dataset, evaluation metrics and related configs for 2.0 (#1663)



* update waymo dataset

* [Fix] Fix all unittests and refactor tests directory and add circle ci in `test-1.x` (#1654)

* add circle ci

* delete github ci

* fix ci

* fix ut

* fix markdown version

* rm

* fix part of uts

* fix comments

* change foler

* refactor test directory

* fix kitti metric ut

* fix all ut
Co-authored-by: default avatarVVsssssk <shenkun@pjlab.org.cn>

* add waymo dataset and evaluation metrics

* convert second configs for v2.0

* [Refactor] Unify ceph config (#1677)

* refactor ceph in config

* support metric load ann file from ceph

* add doc string and remove useless code

* [Fix]Fix create data (#1659)

* add circle ci

* delete github ci

* fix ci

* fix ut

* fix markdown version

* rm

* fix part of uts

* fix comments

* change foler

* refactor test directory

* fix kitti metric ut

* fix all ut

* fix creat data
Co-authored-by: default avatarChaimZhu <zhuchenming@pjlab.org.cn>

* [Fix] Fix seg mapping (#1681)

* [Doc]: fix markdown version (#1653)

* [CI] Add circle ci (#1647)

* add circle ci

* delete github ci

* fix ci

* fix ut

* fix markdown version

* rm

* fix seg mapping for scannet

* fix requiremetn

* fix all seg dataet

* resolve commnets
Co-authored-by: default avatarVVsssssk <88368822+VVsssssk@users.noreply.github.com>

* [Fix] Fix SSN configs (#1686)

* modify doc string and evaluation file location

* add doc string

* remove path mapping in flieclient args
Co-authored-by: default avatarChaimZhu <zhuchenming@pjlab.org.cn>
Co-authored-by: default avatarVVsssssk <shenkun@pjlab.org.cn>
Co-authored-by: default avatarVVsssssk <88368822+VVsssssk@users.noreply.github.com>
Co-authored-by: default avatarShilong Zhang <61961338+jshilong@users.noreply.github.com>
parent a8f3ec5f
...@@ -121,7 +121,9 @@ db_sampler = dict( ...@@ -121,7 +121,9 @@ db_sampler = dict(
Car=15, Car=15,
Pedestrian=15, Pedestrian=15,
Cyclist=15, Cyclist=15,
)) ),
points_loader=dict(
type='LoadPointsFromFile', coord_type='LIDAR', load_dim=4, use_dim=4))
train_pipeline = [ train_pipeline = [
dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=4, use_dim=4), dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=4, use_dim=4),
......
...@@ -111,23 +111,16 @@ db_sampler = dict( ...@@ -111,23 +111,16 @@ db_sampler = dict(
Car=20, Car=20,
Pedestrian=15, Pedestrian=15,
Cyclist=15, Cyclist=15,
)) ),
points_loader=dict(
type='LoadPointsFromFile', coord_type='LIDAR', load_dim=4, use_dim=4))
file_client_args = dict(backend='disk') file_client_args = dict(backend='disk')
# file_client_args = dict( # file_client_args = dict(
# backend='petrel', path_mapping=dict(data='s3://kitti_data/')) # backend='petrel', path_mapping=dict(data='s3://kitti_data/'))
train_pipeline = [ train_pipeline = [
dict( dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=4, use_dim=4),
type='LoadPointsFromFile', dict(type='LoadAnnotations3D', with_bbox_3d=True, with_label_3d=True),
coord_type='LIDAR',
load_dim=4,
use_dim=4,
file_client_args=file_client_args),
dict(
type='LoadAnnotations3D',
with_bbox_3d=True,
with_label_3d=True,
file_client_args=file_client_args),
dict(type='ObjectSample', db_sampler=db_sampler), dict(type='ObjectSample', db_sampler=db_sampler),
dict(type='RandomFlip3D', flip_ratio_bev_horizontal=0.5), dict(type='RandomFlip3D', flip_ratio_bev_horizontal=0.5),
dict( dict(
...@@ -141,12 +134,7 @@ train_pipeline = [ ...@@ -141,12 +134,7 @@ train_pipeline = [
dict(type='Collect3D', keys=['points', 'gt_bboxes_3d', 'gt_labels_3d']) dict(type='Collect3D', keys=['points', 'gt_bboxes_3d', 'gt_labels_3d'])
] ]
test_pipeline = [ test_pipeline = [
dict( dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=4, use_dim=4),
type='LoadPointsFromFile',
coord_type='LIDAR',
load_dim=4,
use_dim=4,
file_client_args=file_client_args),
dict( dict(
type='MultiScaleFlipAug3D', type='MultiScaleFlipAug3D',
img_scale=(1333, 800), img_scale=(1333, 800),
...@@ -171,12 +159,7 @@ test_pipeline = [ ...@@ -171,12 +159,7 @@ test_pipeline = [
# construct a pipeline for data and gt loading in show function # construct a pipeline for data and gt loading in show function
# please keep its loading function consistent with test_pipeline (e.g. client) # please keep its loading function consistent with test_pipeline (e.g. client)
eval_pipeline = [ eval_pipeline = [
dict( dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=4, use_dim=4),
type='LoadPointsFromFile',
coord_type='LIDAR',
load_dim=4,
use_dim=4,
file_client_args=file_client_args),
dict( dict(
type='DefaultFormatBundle3D', type='DefaultFormatBundle3D',
class_names=class_names, class_names=class_names,
......
...@@ -62,21 +62,14 @@ db_sampler = dict( ...@@ -62,21 +62,14 @@ db_sampler = dict(
type='LoadPointsFromFile', type='LoadPointsFromFile',
coord_type='LIDAR', coord_type='LIDAR',
load_dim=5, load_dim=5,
use_dim=[0, 1, 2, 3, 4], use_dim=[0, 1, 2, 3, 4]))
file_client_args=file_client_args))
train_pipeline = [ train_pipeline = [
dict( dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=5, use_dim=5),
type='LoadPointsFromFile',
coord_type='LIDAR',
load_dim=5,
use_dim=5,
file_client_args=file_client_args),
dict( dict(
type='LoadPointsFromMultiSweeps', type='LoadPointsFromMultiSweeps',
sweeps_num=9, sweeps_num=9,
use_dim=[0, 1, 2, 3, 4], use_dim=[0, 1, 2, 3, 4],
file_client_args=file_client_args,
pad_empty_sweeps=True, pad_empty_sweeps=True,
remove_close=True), remove_close=True),
dict(type='LoadAnnotations3D', with_bbox_3d=True, with_label_3d=True), dict(type='LoadAnnotations3D', with_bbox_3d=True, with_label_3d=True),
...@@ -101,17 +94,11 @@ train_pipeline = [ ...@@ -101,17 +94,11 @@ train_pipeline = [
keys=['points', 'gt_bboxes_3d', 'gt_labels_3d']) keys=['points', 'gt_bboxes_3d', 'gt_labels_3d'])
] ]
test_pipeline = [ test_pipeline = [
dict( dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=5, use_dim=5),
type='LoadPointsFromFile',
coord_type='LIDAR',
load_dim=5,
use_dim=5,
file_client_args=file_client_args),
dict( dict(
type='LoadPointsFromMultiSweeps', type='LoadPointsFromMultiSweeps',
sweeps_num=9, sweeps_num=9,
use_dim=[0, 1, 2, 3, 4], use_dim=[0, 1, 2, 3, 4],
file_client_args=file_client_args,
pad_empty_sweeps=True, pad_empty_sweeps=True,
remove_close=True), remove_close=True),
dict( dict(
......
...@@ -8,17 +8,11 @@ class_names = [ ...@@ -8,17 +8,11 @@ class_names = [
] ]
test_pipeline = [ test_pipeline = [
dict( dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=5, use_dim=5),
type='LoadPointsFromFile',
coord_type='LIDAR',
load_dim=5,
use_dim=5,
file_client_args=file_client_args),
dict( dict(
type='LoadPointsFromMultiSweeps', type='LoadPointsFromMultiSweeps',
sweeps_num=9, sweeps_num=9,
use_dim=[0, 1, 2, 3, 4], use_dim=[0, 1, 2, 3, 4],
file_client_args=file_client_args,
pad_empty_sweeps=True, pad_empty_sweeps=True,
remove_close=True), remove_close=True),
dict( dict(
......
...@@ -10,17 +10,11 @@ class_names = [ ...@@ -10,17 +10,11 @@ class_names = [
] ]
test_pipeline = [ test_pipeline = [
dict( dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=5, use_dim=5),
type='LoadPointsFromFile',
coord_type='LIDAR',
load_dim=5,
use_dim=5,
file_client_args=file_client_args),
dict( dict(
type='LoadPointsFromMultiSweeps', type='LoadPointsFromMultiSweeps',
sweeps_num=9, sweeps_num=9,
use_dim=[0, 1, 2, 3, 4], use_dim=[0, 1, 2, 3, 4],
file_client_args=file_client_args,
pad_empty_sweeps=True, pad_empty_sweeps=True,
remove_close=True), remove_close=True),
dict( dict(
......
...@@ -9,17 +9,11 @@ class_names = [ ...@@ -9,17 +9,11 @@ class_names = [
] ]
test_pipeline = [ test_pipeline = [
dict( dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=5, use_dim=5),
type='LoadPointsFromFile',
coord_type='LIDAR',
load_dim=5,
use_dim=5,
file_client_args=file_client_args),
dict( dict(
type='LoadPointsFromMultiSweeps', type='LoadPointsFromMultiSweeps',
sweeps_num=9, sweeps_num=9,
use_dim=[0, 1, 2, 3, 4], use_dim=[0, 1, 2, 3, 4],
file_client_args=file_client_args,
pad_empty_sweeps=True, pad_empty_sweeps=True,
remove_close=True), remove_close=True),
dict( dict(
......
...@@ -57,21 +57,14 @@ db_sampler = dict( ...@@ -57,21 +57,14 @@ db_sampler = dict(
type='LoadPointsFromFile', type='LoadPointsFromFile',
coord_type='LIDAR', coord_type='LIDAR',
load_dim=5, load_dim=5,
use_dim=[0, 1, 2, 3, 4], use_dim=[0, 1, 2, 3, 4]))
file_client_args=file_client_args))
train_pipeline = [ train_pipeline = [
dict( dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=5, use_dim=5),
type='LoadPointsFromFile',
coord_type='LIDAR',
load_dim=5,
use_dim=5,
file_client_args=file_client_args),
dict( dict(
type='LoadPointsFromMultiSweeps', type='LoadPointsFromMultiSweeps',
sweeps_num=9, sweeps_num=9,
use_dim=[0, 1, 2, 3, 4], use_dim=[0, 1, 2, 3, 4],
file_client_args=file_client_args,
pad_empty_sweeps=True, pad_empty_sweeps=True,
remove_close=True), remove_close=True),
dict(type='LoadAnnotations3D', with_bbox_3d=True, with_label_3d=True), dict(type='LoadAnnotations3D', with_bbox_3d=True, with_label_3d=True),
...@@ -95,17 +88,11 @@ train_pipeline = [ ...@@ -95,17 +88,11 @@ train_pipeline = [
keys=['points', 'gt_bboxes_3d', 'gt_labels_3d']) keys=['points', 'gt_bboxes_3d', 'gt_labels_3d'])
] ]
test_pipeline = [ test_pipeline = [
dict( dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=5, use_dim=5),
type='LoadPointsFromFile',
coord_type='LIDAR',
load_dim=5,
use_dim=5,
file_client_args=file_client_args),
dict( dict(
type='LoadPointsFromMultiSweeps', type='LoadPointsFromMultiSweeps',
sweeps_num=9, sweeps_num=9,
use_dim=[0, 1, 2, 3, 4], use_dim=[0, 1, 2, 3, 4],
file_client_args=file_client_args,
pad_empty_sweeps=True, pad_empty_sweeps=True,
remove_close=True), remove_close=True),
dict( dict(
...@@ -133,12 +120,11 @@ eval_pipeline = [ ...@@ -133,12 +120,11 @@ eval_pipeline = [
coord_type='LIDAR', coord_type='LIDAR',
load_dim=5, load_dim=5,
use_dim=5, use_dim=5,
file_client_args=file_client_args), ),
dict( dict(
type='LoadPointsFromMultiSweeps', type='LoadPointsFromMultiSweeps',
sweeps_num=9, sweeps_num=9,
use_dim=[0, 1, 2, 3, 4], use_dim=[0, 1, 2, 3, 4],
file_client_args=file_client_args,
pad_empty_sweeps=True, pad_empty_sweeps=True,
remove_close=True), remove_close=True),
dict(type='Pack3DDetInputs', keys=['points']) dict(type='Pack3DDetInputs', keys=['points'])
......
...@@ -58,21 +58,14 @@ db_sampler = dict( ...@@ -58,21 +58,14 @@ db_sampler = dict(
type='LoadPointsFromFile', type='LoadPointsFromFile',
coord_type='LIDAR', coord_type='LIDAR',
load_dim=5, load_dim=5,
use_dim=[0, 1, 2, 3, 4], use_dim=[0, 1, 2, 3, 4]))
file_client_args=file_client_args))
train_pipeline = [ train_pipeline = [
dict( dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=5, use_dim=5),
type='LoadPointsFromFile',
coord_type='LIDAR',
load_dim=5,
use_dim=5,
file_client_args=file_client_args),
dict( dict(
type='LoadPointsFromMultiSweeps', type='LoadPointsFromMultiSweeps',
sweeps_num=9, sweeps_num=9,
use_dim=[0, 1, 2, 3, 4], use_dim=[0, 1, 2, 3, 4],
file_client_args=file_client_args,
pad_empty_sweeps=True, pad_empty_sweeps=True,
remove_close=True), remove_close=True),
dict(type='LoadAnnotations3D', with_bbox_3d=True, with_label_3d=True), dict(type='LoadAnnotations3D', with_bbox_3d=True, with_label_3d=True),
...@@ -96,17 +89,11 @@ train_pipeline = [ ...@@ -96,17 +89,11 @@ train_pipeline = [
keys=['points', 'gt_bboxes_3d', 'gt_labels_3d']) keys=['points', 'gt_bboxes_3d', 'gt_labels_3d'])
] ]
test_pipeline = [ test_pipeline = [
dict( dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=5, use_dim=5),
type='LoadPointsFromFile',
coord_type='LIDAR',
load_dim=5,
use_dim=5,
file_client_args=file_client_args),
dict( dict(
type='LoadPointsFromMultiSweeps', type='LoadPointsFromMultiSweeps',
sweeps_num=9, sweeps_num=9,
use_dim=[0, 1, 2, 3, 4], use_dim=[0, 1, 2, 3, 4],
file_client_args=file_client_args,
pad_empty_sweeps=True, pad_empty_sweeps=True,
remove_close=True), remove_close=True),
dict( dict(
...@@ -127,17 +114,11 @@ test_pipeline = [ ...@@ -127,17 +114,11 @@ test_pipeline = [
# construct a pipeline for data and gt loading in show function # construct a pipeline for data and gt loading in show function
# please keep its loading function consistent with test_pipeline (e.g. client) # please keep its loading function consistent with test_pipeline (e.g. client)
eval_pipeline = [ eval_pipeline = [
dict( dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=5, use_dim=5),
type='LoadPointsFromFile',
coord_type='LIDAR',
load_dim=5,
use_dim=5,
file_client_args=file_client_args),
dict( dict(
type='LoadPointsFromMultiSweeps', type='LoadPointsFromMultiSweeps',
sweeps_num=9, sweeps_num=9,
use_dim=[0, 1, 2, 3, 4], use_dim=[0, 1, 2, 3, 4],
file_client_args=file_client_args,
pad_empty_sweeps=True, pad_empty_sweeps=True,
remove_close=True), remove_close=True),
dict(type='Pack3DDetInputs', keys=['points']) dict(type='Pack3DDetInputs', keys=['points'])
......
...@@ -14,21 +14,21 @@ model = dict( ...@@ -14,21 +14,21 @@ model = dict(
dcn=dict(type='DCNv2', deform_groups=1, fallback_on_stride=False), dcn=dict(type='DCNv2', deform_groups=1, fallback_on_stride=False),
stage_with_dcn=(False, False, True, True))) stage_with_dcn=(False, False, True, True)))
# file_client_args = dict(backend='disk') file_client_args = dict(backend='disk')
# Uncomment the following if use ceph or other file clients. # Uncomment the following if use ceph or other file clients.
# See https://mmcv.readthedocs.io/en/latest/api.html#mmcv.fileio.FileClient # See https://mmcv.readthedocs.io/en/latest/api.html#mmcv.fileio.FileClient
# for more details. # for more details.
file_client_args = dict( # file_client_args = dict(
backend='petrel', # backend='petrel',
path_mapping=dict({ # path_mapping=dict({
'./data/nuscenes/': # './data/nuscenes/':
's3://openmmlab/datasets/detection3d/nuscenes/', # 's3://openmmlab/datasets/detection3d/nuscenes/',
'data/nuscenes/': # 'data/nuscenes/':
's3://openmmlab/datasets/detection3d/nuscenes/' # 's3://openmmlab/datasets/detection3d/nuscenes/'
})) # }))
train_pipeline = [ train_pipeline = [
dict(type='LoadImageFromFileMono3D', file_client_args=file_client_args), dict(type='LoadImageFromFileMono3D'),
dict( dict(
type='LoadAnnotations3D', type='LoadAnnotations3D',
with_bbox=True, with_bbox=True,
...@@ -47,7 +47,7 @@ train_pipeline = [ ...@@ -47,7 +47,7 @@ train_pipeline = [
]), ]),
] ]
test_pipeline = [ test_pipeline = [
dict(type='LoadImageFromFileMono3D', file_client_args=file_client_args), dict(type='LoadImageFromFileMono3D'),
dict(type='mmdet.Resize', scale_factor=1.0), dict(type='mmdet.Resize', scale_factor=1.0),
dict(type='Pack3DDetInputs', keys=['img']) dict(type='Pack3DDetInputs', keys=['img'])
] ]
......
...@@ -36,16 +36,8 @@ file_client_args = dict(backend='disk') ...@@ -36,16 +36,8 @@ file_client_args = dict(backend='disk')
# 'data/nuscenes/': 's3://nuscenes/nuscenes/' # 'data/nuscenes/': 's3://nuscenes/nuscenes/'
# })) # }))
train_pipeline = [ train_pipeline = [
dict( dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=5, use_dim=5),
type='LoadPointsFromFile', dict(type='LoadPointsFromMultiSweeps', sweeps_num=10),
coord_type='LIDAR',
load_dim=5,
use_dim=5,
file_client_args=file_client_args),
dict(
type='LoadPointsFromMultiSweeps',
sweeps_num=10,
file_client_args=file_client_args),
dict(type='LoadAnnotations3D', with_bbox_3d=True, with_label_3d=True), dict(type='LoadAnnotations3D', with_bbox_3d=True, with_label_3d=True),
dict( dict(
type='GlobalRotScaleTrans', type='GlobalRotScaleTrans',
......
...@@ -94,10 +94,7 @@ train_pipeline = [ ...@@ -94,10 +94,7 @@ train_pipeline = [
with_mask_3d=True, with_mask_3d=True,
with_seg_3d=True), with_seg_3d=True),
dict(type='GlobalAlignment', rotation_axis=2), dict(type='GlobalAlignment', rotation_axis=2),
dict( dict(type='PointSegClassMapping'),
type='PointSegClassMapping',
valid_cat_ids=(3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 14, 16, 24, 28, 33, 34,
36, 39)),
dict(type='PointSample', num_points=50000), dict(type='PointSample', num_points=50000),
dict( dict(
type='RandomFlip3D', type='RandomFlip3D',
......
...@@ -93,10 +93,7 @@ train_pipeline = [ ...@@ -93,10 +93,7 @@ train_pipeline = [
with_mask_3d=True, with_mask_3d=True,
with_seg_3d=True), with_seg_3d=True),
dict(type='GlobalAlignment', rotation_axis=2), dict(type='GlobalAlignment', rotation_axis=2),
dict( dict(type='PointSegClassMapping'),
type='PointSegClassMapping',
valid_cat_ids=(3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 14, 16, 24, 28, 33, 34,
36, 39)),
dict(type='PointSample', num_points=50000), dict(type='PointSample', num_points=50000),
dict( dict(
type='RandomFlip3D', type='RandomFlip3D',
......
...@@ -109,10 +109,7 @@ train_pipeline = [ ...@@ -109,10 +109,7 @@ train_pipeline = [
with_mask_3d=True, with_mask_3d=True,
with_seg_3d=True), with_seg_3d=True),
dict(type='GlobalAlignment', rotation_axis=2), dict(type='GlobalAlignment', rotation_axis=2),
dict( dict(type='PointSegClassMapping'),
type='PointSegClassMapping',
valid_cat_ids=(3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 14, 16, 24, 28, 33, 34,
36, 39)),
dict(type='PointSample', num_points=50000), dict(type='PointSample', num_points=50000),
dict( dict(
type='RandomFlip3D', type='RandomFlip3D',
......
...@@ -110,10 +110,7 @@ train_pipeline = [ ...@@ -110,10 +110,7 @@ train_pipeline = [
with_mask_3d=True, with_mask_3d=True,
with_seg_3d=True), with_seg_3d=True),
dict(type='GlobalAlignment', rotation_axis=2), dict(type='GlobalAlignment', rotation_axis=2),
dict( dict(type='PointSegClassMapping'),
type='PointSegClassMapping',
valid_cat_ids=(3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 14, 16, 24, 28, 33, 34,
36, 39)),
dict(type='PointSample', num_points=50000), dict(type='PointSample', num_points=50000),
dict( dict(
type='RandomFlip3D', type='RandomFlip3D',
......
...@@ -94,7 +94,7 @@ file_client_args = dict( ...@@ -94,7 +94,7 @@ file_client_args = dict(
train_pipeline = [ train_pipeline = [
dict(type='LoadAnnotations3D'), dict(type='LoadAnnotations3D'),
dict(type='LoadImageFromFileMono3D', file_client_args=file_client_args), dict(type='LoadImageFromFileMono3D'),
dict(type='RandomFlip3D', flip_ratio_bev_horizontal=0.5), dict(type='RandomFlip3D', flip_ratio_bev_horizontal=0.5),
dict( dict(
type='RandomResize', scale=[(1173, 352), (1387, 416)], type='RandomResize', scale=[(1173, 352), (1387, 416)],
...@@ -103,7 +103,7 @@ train_pipeline = [ ...@@ -103,7 +103,7 @@ train_pipeline = [
dict(type='Pack3DDetInputs', keys=['img', 'gt_bboxes_3d', 'gt_labels_3d']) dict(type='Pack3DDetInputs', keys=['img', 'gt_bboxes_3d', 'gt_labels_3d'])
] ]
test_pipeline = [ test_pipeline = [
dict(type='LoadImageFromFileMono3D', file_client_args=file_client_args), dict(type='LoadImageFromFileMono3D'),
dict(type='Resize', scale=(1280, 384), keep_ratio=True), dict(type='Resize', scale=(1280, 384), keep_ratio=True),
dict(type='Pack3DDetInputs', keys=['img']) dict(type='Pack3DDetInputs', keys=['img'])
] ]
......
...@@ -17,7 +17,7 @@ img_norm_cfg = dict( ...@@ -17,7 +17,7 @@ img_norm_cfg = dict(
mean=[123.675, 116.28, 103.53], std=[58.395, 57.12, 57.375], to_rgb=True) mean=[123.675, 116.28, 103.53], std=[58.395, 57.12, 57.375], to_rgb=True)
test_pipeline = [ test_pipeline = [
dict(type='LoadImageFromFile', file_client_args=file_client_args), dict(type='LoadImageFromFile'),
dict( dict(
type='MultiScaleFlipAug', type='MultiScaleFlipAug',
img_scale=(1600, 900), img_scale=(1600, 900),
......
...@@ -22,10 +22,7 @@ train_pipeline = [ ...@@ -22,10 +22,7 @@ train_pipeline = [
with_label_3d=False, with_label_3d=False,
with_mask_3d=False, with_mask_3d=False,
with_seg_3d=True), with_seg_3d=True),
dict( dict(type='PointSegClassMapping'),
type='PointSegClassMapping',
valid_cat_ids=tuple(range(len(class_names))),
max_cat_id=13),
dict( dict(
type='IndoorPatchPointSample', type='IndoorPatchPointSample',
num_points=num_points, num_points=num_points,
......
...@@ -26,15 +26,13 @@ train_pipeline = [ ...@@ -26,15 +26,13 @@ train_pipeline = [
shift_height=False, shift_height=False,
use_color=True, use_color=True,
load_dim=6, load_dim=6,
use_dim=[0, 1, 2, 3, 4, 5], use_dim=[0, 1, 2, 3, 4, 5]),
file_client_args=file_client_args),
dict( dict(
type='LoadAnnotations3D', type='LoadAnnotations3D',
with_bbox_3d=False, with_bbox_3d=False,
with_label_3d=False, with_label_3d=False,
with_mask_3d=False, with_mask_3d=False,
with_seg_3d=True, with_seg_3d=True),
file_client_args=file_client_args),
dict(type='PointSegClassMapping'), dict(type='PointSegClassMapping'),
dict( dict(
type='IndoorPatchPointSample', type='IndoorPatchPointSample',
......
...@@ -18,7 +18,9 @@ db_sampler = dict( ...@@ -18,7 +18,9 @@ db_sampler = dict(
filter_by_difficulty=[-1], filter_by_difficulty=[-1],
filter_by_min_points=dict(Car=5, Pedestrian=10, Cyclist=10)), filter_by_min_points=dict(Car=5, Pedestrian=10, Cyclist=10)),
classes=class_names, classes=class_names,
sample_groups=dict(Car=12, Pedestrian=6, Cyclist=6)) sample_groups=dict(Car=12, Pedestrian=6, Cyclist=6),
points_loader=dict(
type='LoadPointsFromFile', coord_type='LIDAR', load_dim=4, use_dim=4))
train_pipeline = [ train_pipeline = [
dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=4, use_dim=4), dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=4, use_dim=4),
dict(type='LoadAnnotations3D', with_bbox_3d=True, with_label_3d=True), dict(type='LoadAnnotations3D', with_bbox_3d=True, with_label_3d=True),
......
...@@ -82,7 +82,9 @@ db_sampler = dict( ...@@ -82,7 +82,9 @@ db_sampler = dict(
rate=1.0, rate=1.0,
prepare=dict(filter_by_difficulty=[-1], filter_by_min_points=dict(Car=5)), prepare=dict(filter_by_difficulty=[-1], filter_by_min_points=dict(Car=5)),
classes=class_names, classes=class_names,
sample_groups=dict(Car=15)) sample_groups=dict(Car=15),
points_loader=dict(
type='LoadPointsFromFile', coord_type='LIDAR', load_dim=4, use_dim=4))
train_pipeline = [ train_pipeline = [
dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=4, use_dim=4), dict(type='LoadPointsFromFile', coord_type='LIDAR', load_dim=4, use_dim=4),
dict(type='LoadAnnotations3D', with_bbox_3d=True, with_label_3d=True), dict(type='LoadAnnotations3D', with_bbox_3d=True, with_label_3d=True),
......
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