search

WongKinYiu / yolor

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)
GNU General Public License v3.0
1.98k stars 524 forks source link

Replacing Yolor backbone #187

Open JoseIgnDiazVilla opened 2 years ago

JoseIgnDiazVilla commented 2 years ago

Hello!

I'm trying to do some tests by replacing the backbone of yolor with some other architectures. Are there any guidelines, advice, or suggestions that I could follow?

Thanks!

joaodinislopes commented 2 years ago

I think the backbone is all defined in the .cfg files in the cfg folder. If you want you can edit the layers there, in those files they even added the separator "# ============ Backbone ============ #" to make it more obvious