minor fixes

darrenjkt · Sep 19, 2023 · 18dd1e7 · 18dd1e7
1 parent a4d7337
commit 18dd1e7
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 11 deletions.
diff --git 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
@@ -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
@@ -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