LeapLabTHU / ARC

[ICCV 2023] Adaptive Rotated Convolution for Rotated Object Detection
Apache License 2.0
111 stars 4 forks source link

pretrained backbone #13

Closed wind-waves-fll closed 5 months ago

wind-waves-fll commented 5 months ago

I want to request how to use pretrained backbone? At the test code,I can see the load_checkpoint to load the model. But when training, can I how to use pretrained backbone?

wind-waves-fll commented 5 months ago

I find the location! in the defalt_runtime.py, set the Load_from= pretrained backbone

yifanpu001 commented 5 months ago

Hi, I think set the pretrained backbone dir in the model config file is a more elegent usage

https://github.com/LeapLabTHU/ARC/blob/4825acd8d5777d668ea4fb3b4d24aa53399d79ff/configs/obb/arc/arc_orcnn_r50fpn1x_ss_dota10_RxFFF_n4.py#L6

wind-waves-fll commented 5 months ago

Hi, I think set the pretrained backbone dir in the model config file is a more elegent usage

https://github.com/LeapLabTHU/ARC/blob/4825acd8d5777d668ea4fb3b4d24aa53399d79ff/configs/obb/arc/arc_orcnn_r50fpn1x_ss_dota10_RxFFF_n4.py#L6

Ok,thank you very much!