Python create_nuscenes_infosの例

コード例 #1

ファイル: create_data.py プロジェクト: zxduan90/CenterPoint

def nuscenes_data_prep(root_path, version, nsweeps=10, filter_zero=True):
    nu_ds.create_nuscenes_infos(root_path, version=version, nsweeps=nsweeps, filter_zero=filter_zero)
    create_groundtruth_database(
        "NUSC",
        root_path,
        Path(root_path) / "infos_train_{:02d}sweeps_withvelo.pkl".format(nsweeps),
        nsweeps=nsweeps,
    )

コード例 #2

def nuscenes_data_prep(root_path,
                       version,
                       nsweeps=10,
                       filter_zero=True,
                       virtual=False):
    nu_ds.create_nuscenes_infos(root_path,
                                version=version,
                                nsweeps=nsweeps,
                                filter_zero=filter_zero)
    if version == 'v1.0-trainval':
        create_groundtruth_database(
            "NUSC",
            root_path,
            Path(root_path) /
            "infos_train_{:02d}sweeps_withvelo_filter_{}.pkl".format(
                nsweeps, filter_zero),
            nsweeps=nsweeps,
            virtual=virtual)

コード例 #3

def nuscenes_data_prep(root_path, version, nsweeps=10):
    nu_ds.create_nuscenes_infos(root_path, version=version, nsweeps=nsweeps)