tusen-ai/SST

Here is my step trying to reproduce CTRL,I want to know is there any wrong ?

Open

#161 aperta il 21 set 2023

Vedi su GitHub
 (12 commenti) (8 reazioni) (0 assegnatari)Python (106 fork)github user discovery
good first issue

Metriche repository

Star
 (883 star)
Metriche merge PR
 (Metriche PR in attesa)

Descrizione

Here is my step trying to reproduce CTRL,I want to know is there any wrong ? especially step2,and Is the config file ‘fsd_base_vehicle.yaml’ correct?

1.prepare waymo data(I only use part of waymo dataset) 1.1 use my python script to generate train.txt val.txt test.txt and idx2timestamp.pkl idx2contextname.pkl Then cp train.txt val.txt test.txt to ./data/waymo/kitti_format/ImageSets/ cp idx2timestamp.pkl idx2contextname.pkl to ./data/waymo/kitti_format/ 1.2 python tools/create_data.py --dataset waymo --root-path ./data/waymo/ --out-dir ./data/waymo/ --workers 128 --extra-tag waymo image Step 1: Generate train_gt.bin once for all. (waymo bin format). python ./tools/ctrl/generate_train_gt_bin.py generate file 'train_gt.bin' image python ./tools/ctrl/extract_poses.py Generate file context2timestamp.pkl and pose.pkl image

Step 2: Use ImmortalTracker to generate tracking results in training split (bin file format) modify file ego_info.py and time_stamp.py like this: image Modify file waymo_convert_detection.sh like this: image then: bash preparedata/waymo/waymo_preparedata.sh ~/dataset/waymo/waymo_format/ generate files like this : image

bash preparedata/waymo/waymo_convert_detection.sh ~/dataset/waymo/waymo_format/train_gt.bin CTRL_FSD_TTA Generate files like this: In data/waymo/training/detection/CTRL_FSD_TTA/dets: image Modify file run_mot.sh like this: image

Then: bash run_mot.sh generate file like this: image Step 3: Generate track input for training modify file ‘fsd_base_vehicle.yaml’ like this: pred.bin was generated in step 2. image python ./tools/ctrl/generate_track_input.py ./tools/ctrl/data_configs/fsd_base_vehicle.yaml --process 1 generate files like this: image

Step 4: Assign candidates GT tracks python ./tools/ctrl/generate_candidates.py ./tools/ctrl/data_configs/fsd_base_vehicle.yaml --process 1

image Step 5: Begin training bash tools/dist_train.sh configs/ctrl/ctrl_veh_24e.py 1 --no-validate

Originally posted by @20210726 in https://github.com/tusen-ai/SST/issues/132#issuecomment-1688148061

Guida contributor