duanzhiihao / RAPiD

RAPiD: Rotation-Aware People Detection in Overhead Fisheye Images (CVPR 2020 Workshops)
http://vip.bu.edu/rapid/
Other
213 stars 63 forks source link

Fine-tune on custom fisheye dataset #39

Closed aparico closed 2 years ago

aparico commented 2 years ago

@duanzhiihao

I would like to fine-tune COCO-pretrained RAPiD on some custom fisheye dataset that is similar to CEPDOF, WEPDTOF, MW-R, and HABBOF.

Just to clarify, what I mean by fine-tune is to freeze the backbone and train only the neck and the head at a lower learning rate.

Could you please upload your fine-tuning functions / script? 🙏

Thank you.

aparico commented 2 years ago

I just wrote the code myself using this as a guide.