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.99k stars 520 forks source link

About num of filters in .cfg file #25

Closed e4exp closed 3 years ago

e4exp commented 3 years ago

Hello, I'm trying to train YOLOR on my custom data.

In previous versions of YOLO, we set (classes + 5) * 3 filters in [convolution] layers before [yolo] layers. I think we should do the same thing to 4 [implicit_mul] layers in yolor_p6.cfg in your work. Is this collect?

Thank you.

WongKinYiu commented 3 years ago

yes, you are correct.

e4exp commented 3 years ago

Thank you for your quick response, I understood.

qutyyds commented 2 years ago

YOLORp6.yaml 中怎么修改呢？