search

uzh-rpg / svit

Official implementation of "SViT: Revisiting Token Pruning for Object Detection and Instance Segmentation"
Apache License 2.0
23 stars 4 forks source link

Utilize lightweight detection head #12

Open King4819 opened 1 month ago

King4819 commented 1 month ago

Hi, I would like to ask that how to modify the code to utilize lightweight detection head like yolos, thanks. Hopes to get your reply !!!

kaikai23 commented 1 month ago

Hi, I haven't tried changing detection heads, but if you want, maybe you could refer to this config.

King4819 commented 1 month ago

Thanks for your reply. I want to ask that have you test using ssd300 as detector ? Since when I replace the config file, it gets many bugs, hopes to get your reply, thanks !!!