Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Code in the kitti_dataset_mm.py "def generate_prediction_dicts"? #46

Closed
weilli opened this issue Sep 15, 2023 · 1 comment
Closed

Code in the kitti_dataset_mm.py "def generate_prediction_dicts"? #46

weilli opened this issue Sep 15, 2023 · 1 comment

Comments

@weilli
Copy link

weilli commented Sep 15, 2023

Sorry to bother you.
I still can't figure it out that why "pred_dict['bbox'][height_mask, 3] +=2"? Are there any details I missed?
Have a nice day!

line 315 to 318 in kitti_dataset_mm.py:
pred_dict['bbox'] = pred_boxes_img
height = pred_dict['bbox'][:, 3] - pred_dict['bbox'][:, 1]
height_mask = height<25
pred_dict['bbox'][height_mask, 3] +=2

@hailanyi
Copy link
Owner

hailanyi commented Dec 7, 2023

Just to keep more objects during evaluation, as the KITTI filters objects of height<25.

@hailanyi hailanyi closed this as completed Dec 7, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants