diff --git a/2_ssl_pretrain.bash b/2_ssl_pretrain.bash index a710e01..76ebdfd 100644 --- a/2_ssl_pretrain.bash +++ b/2_ssl_pretrain.bash @@ -16,6 +16,7 @@ SETTING=lidar_only_weakly_supervised # 3. --------------MAE method-------------- METHOD=ssl_pointpillar_single_mae FUSION_METHOD=late +bfksbifb # single GPU CUDA_VISIBLE_DEVICES=0 python opencood/tools/train_mae.py \ diff --git a/3_train_static_teacher.bash b/3_train_static_teacher.bash index c707a21..61200e4 100644 --- a/3_train_static_teacher.bash +++ b/3_train_static_teacher.bash @@ -22,7 +22,7 @@ SETTING=lidar_only_weakly_supervised METHOD=pointpillar_discograph # METHOD=pointpillar_v2vnet # METHOD=pointpillar_v2xvit - +jkafksbfib # 4.--------------fusion scheme-------------- # FUSION_METHOD=late # FUSION_METHOD=early diff --git a/5_train_HINTED.bash b/5_train_HINTED.bash index 5403bbe..17b45a2 100644 --- a/5_train_HINTED.bash +++ b/5_train_HINTED.bash @@ -19,7 +19,7 @@ METHOD=discograph # METHOD=v2vnet YAML_FILE=opencood/hypes_yaml/$DATASET/$SETTING/pointpillar_${METHOD}.yaml - +sajfboauefb # single GPU: finetune,v2xsim, dair_v2x TEACHER_MODEL=${METHOD}_ws_single_mae_0.7 # 第一步 diff --git a/data_preparation/v2xsim_prepare.py b/data_preparation/v2xsim_prepare.py index 02412bd..6d03262 100644 --- a/data_preparation/v2xsim_prepare.py +++ b/data_preparation/v2xsim_prepare.py @@ -16,7 +16,7 @@ from opencood.utils.box_utils import corner_to_center, mask_boxes_outside_range_numpy, create_bbx, get_points_in_rotated_box_3d from pdb import set_trace as pause - + # dhathri demo class def quaternion_yaw(q: Quaternion) -> float: """ Calculate the yaw angle from a quaternion.