tusen-ai/SST

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

Open

#161 geöffnet am 21. Sept. 2023

Auf GitHub ansehen
 (12 Kommentare) (8 Reaktionen) (0 zugewiesene Personen)Python (106 Forks)github user discovery
good first issue

Repository-Metriken

Stars
 (883 Stars)
PR-Merge-Metriken
 (PR-Metriken ausstehend)

Beschreibung

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

Contributor Guide