Hzzone / PseCo

(CVPR 2024) Point, Segment and Count: A Generalized Framework for Object Counting
89 stars 7 forks source link

About pseudo_boxes_data_vith.pth #20

Closed XiruiTeng closed 3 hours ago

XiruiTeng commented 4 days ago

Hello, I appreciate your great work! When running the file 4_1_train_roi_head.py, I noticed that the code requires a file named pseudo_boxes_data_vith.pth, which I couldn't find in the previous steps. I'm wondering if the 'pred_boxes' element in this file is the same as the 'pred_boxes' in all_predictions or in segment_anything_data?

Hzzone commented 3 days ago

https://github.com/Hzzone/PseCo/blob/main/fsc147/1_generate_data.ipynb Generate pseudo labels with ground-truth point prompts