search

yangxue0827 / RotationDetection

This is a tensorflow-based rotation detection benchmark, also called AlphaRotate.
https://rotationdetection.readthedocs.io/
Apache License 2.0
1.08k stars 181 forks source link

Tranfer Learning #108

Closed wafa-bouzouita closed 1 year ago

wafa-bouzouita commented 2 years ago

Hello, I have a question regarding training a custom dataset.

How can I transfer learning of some specific classes from the pre-trained weights (e.g. dota) to my custom training if my custom classes are different from the pre-trained classes?

Best regards and Thank you

yangxue0827 commented 2 years ago

Filter the last layer when loading weights.