train new model ---yolor-p6.yaml

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

train new model ---yolor-p6.yaml #253

Open qutyyds opened 2 years ago

qutyyds commented 2 years ago

If I want to train a new model with yolor-p6, what do I need to do besides modifying "number of classes"? I didn't find a place to modify "filters=255" like yolor-p6.cfg. What should I do? Can you help me? QQ截图20220615212055 QQ截图20220615212706

Splendon commented 2 years ago

hi @qutyyds, you can refer to this github project, https://github.com/DataXujing/YOLOR-

qutyyds commented 2 years ago

你好@qutyyds，可以参考这个github项目，DataXjing/YOLOR-

Thanks for your answer, but it doesn't solve my problem. I want to change ""yolor-p6.yaml"" but not "yolo-p6.cfg".