From 18dd1e75ccbf4ee0d362049edefa4624db2f8218 Mon Sep 17 00:00:00 2001 From: Darren Date: Wed, 20 Sep 2023 07:55:20 +1000 Subject: [PATCH] minor fixes --- pcdet/datasets/custom/custom_dataset.py | 9 ++------- .../round2/scripts/generate_ensemble_preds.sh | 4 ++-- .../round3/scripts/generate_ensemble_preds.sh | 4 ++-- 3 files changed, 6 insertions(+), 11 deletions(-) diff --git a/pcdet/datasets/custom/custom_dataset.py b/pcdet/datasets/custom/custom_dataset.py index 1cf31b7..cb1828a 100644 --- a/pcdet/datasets/custom/custom_dataset.py +++ b/pcdet/datasets/custom/custom_dataset.py @@ -2,12 +2,8 @@ import pickle import numpy as np -from skimage import io import open3d as o3d -import json -from ...ops.roiaware_pool3d import roiaware_pool3d_utils -from ...utils import box_utils, calibration_kitti, common_utils, object3d_kitti from ..dataset import DatasetTemplate class CustomDataset(DatasetTemplate): @@ -41,6 +37,8 @@ def include_data(self): self.seq_name_to_len[infos[0]['point_cloud']['lidar_sequence']] = len(infos) self.infos.extend(custom_infos) + if self.logger is not None: + self.logger.info('Total samples for CustomDataset dataset: %d' % (len(self.infos))) if self.dataset_cfg.SAMPLED_INTERVAL[self.mode] > 1: sampled_infos = [] @@ -60,9 +58,6 @@ def include_data(self): for idx, data in enumerate(self.infos): self.frameid_to_idx[data['frame_id']] = idx - if self.logger is not None: - self.logger.info('Total samples for CustomDataset dataset: %d' % (len(self.infos))) - return seq_name_to_infos def __len__(self): diff --git a/tools/cfgs/target_nuscenes/label_generation/round2/scripts/generate_ensemble_preds.sh b/tools/cfgs/target_nuscenes/label_generation/round2/scripts/generate_ensemble_preds.sh index 1f81c81..5388036 100644 --- a/tools/cfgs/target_nuscenes/label_generation/round2/scripts/generate_ensemble_preds.sh +++ b/tools/cfgs/target_nuscenes/label_generation/round2/scripts/generate_ensemble_preds.sh @@ -16,14 +16,14 @@ python test.py --cfg_file cfgs/target_nuscenes/ms3d_waymo_voxel_rcnn_centerhead. python test.py --cfg_file cfgs/target_nuscenes/ms3d_waymo_voxel_rcnn_centerhead.yaml \ --ckpt ../output/target_nuscenes/ms3d_waymo_voxel_rcnn_centerhead/10f_xyzt_vehped_rnd1/ckpt/checkpoint_epoch_30.pth \ - --eval_tag nusc10xyzt_rwr \ + --eval_tag nusc10xyzt_rwf \ --extra_tag 10f_xyzt_vehped_rnd1 \ --target_dataset nuscenes --sweeps 10 --use_tta 1 --custom_target_scenes \ --set DATA_CONFIG_TAR.DATA_SPLIT.test train MODEL.POST_PROCESSING.EVAL_METRIC none python test.py --cfg_file cfgs/target_nuscenes/ms3d_waymo_voxel_rcnn_centerhead.yaml \ --ckpt ../output/target_nuscenes/ms3d_waymo_voxel_rcnn_centerhead/10f_xyzt_vehped_rnd1/ckpt/checkpoint_epoch_30.pth \ - --eval_tag nusc10xyzt_rwr \ + --eval_tag nusc10xyzt_rwf_rwr \ --extra_tag 10f_xyzt_vehped_rnd1 \ --target_dataset nuscenes --sweeps 10 --use_tta 3 --custom_target_scenes \ --set DATA_CONFIG_TAR.DATA_SPLIT.test train MODEL.POST_PROCESSING.EVAL_METRIC none diff --git a/tools/cfgs/target_nuscenes/label_generation/round3/scripts/generate_ensemble_preds.sh b/tools/cfgs/target_nuscenes/label_generation/round3/scripts/generate_ensemble_preds.sh index ef00bfe..a0b3ab1 100644 --- a/tools/cfgs/target_nuscenes/label_generation/round3/scripts/generate_ensemble_preds.sh +++ b/tools/cfgs/target_nuscenes/label_generation/round3/scripts/generate_ensemble_preds.sh @@ -16,14 +16,14 @@ python test.py --cfg_file cfgs/target_nuscenes/ms3d_waymo_voxel_rcnn_centerhead. python test.py --cfg_file cfgs/target_nuscenes/ms3d_waymo_voxel_rcnn_centerhead.yaml \ --ckpt ../output/target_nuscenes/ms3d_waymo_voxel_rcnn_centerhead/10f_xyzt_vehped_rnd2/ckpt/checkpoint_epoch_30.pth \ - --eval_tag nusc10xyzt_rwr \ + --eval_tag nusc10xyzt_rwf \ --extra_tag 10f_xyzt_vehped_rnd2 \ --target_dataset nuscenes --sweeps 10 --use_tta 1 --custom_target_scenes \ --set DATA_CONFIG_TAR.DATA_SPLIT.test train MODEL.POST_PROCESSING.EVAL_METRIC none python test.py --cfg_file cfgs/target_nuscenes/ms3d_waymo_voxel_rcnn_centerhead.yaml \ --ckpt ../output/target_nuscenes/ms3d_waymo_voxel_rcnn_centerhead/10f_xyzt_vehped_rnd2/ckpt/checkpoint_epoch_30.pth \ - --eval_tag nusc10xyzt_rwr \ + --eval_tag nusc10xyzt_rwf_rwr \ --extra_tag 10f_xyzt_vehped_rnd2 \ --target_dataset nuscenes --sweeps 10 --use_tta 3 --custom_target_scenes \ --set DATA_CONFIG_TAR.DATA_SPLIT.test train MODEL.POST_PROCESSING.EVAL_METRIC none