chensnathan / YOLOF

You Only Look One-level Feature (YOLOF), CVPR2021, Detectron2
MIT License
271 stars 28 forks source link

A problem about the code at train_net.py ? #23

Closed randomNNN closed 3 years ago

randomNNN commented 3 years ago

Dear author,I have had a problem when I saw the code:

optimizer = torch.optim.SGD( params, cfg.SOLVER.BASE_LR, momentum=cfg.SOLVER.MOMENTUM )

where you aren't define the cfg.SOLVER.MOMENTUM in the configs.Appreciate!

chensnathan commented 3 years ago

Hi, we adopt many default settings in Detectron2. And we only give the modification configs in this repo. Detailed setting for SOLVER.MOMENTUM can be found here.

randomNNN commented 3 years ago

Thank for your reply.When I first saw your paper,the YOLOF's network architecture really surprised me,the siso encoder and uniform matcher are brilliant.At the same time, I have an another problem,if I wanna instead the backbone of mobilenetv3,and use the CA attention in the backbone.Can the OD model work well?Appreciate!------------------ 原始邮件 ------------------ @.> 发送时间: 2021年6月17日(星期四) 下午5:17 @.>; @.**@.>; 主题: Re: [chensnathan/YOLOF] A problem about the code at train_net.py ? (#23)

chensnathan commented 3 years ago

Sorry for the late reply.

YOLOF can definitely work well when you change a new backbone. You can try it with mobilenetv3.

randomNNN commented 3 years ago

Thanks for your reply.I will try to instead the backbone of lightwight model.And I have a presumptuous request.Can you give me a weChat、QQ  or mailbox to communicate more detail.It's ok to refuse if this would bother you.Appreciate!

------------------ 原始邮件 ------------------ 发件人: "qiang @.>; 发送时间: 2021年6月20日(星期天) 晚上7:34 收件人: @.>; 抄送: @.>; @.>; 主题: Re: [chensnathan/YOLOF] A problem about the code at train_net.py ? (#23)

Sorry for the late reply.

YOLOF can definitely work well when you change a new backbone. You can try it with mobilenetv3.

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.

chensnathan commented 3 years ago

qiang.chen@nlpr.ia.ac.cn

If you have any questions, feel free to drop me an email.

randomNNN commented 3 years ago

Thanks!

------------------ 原始邮件 ------------------ 发件人: "qiang @.>; 发送时间: 2021年6月21日(星期一) 下午5:18 收件人: @.>; 抄送: @.>; @.>; 主题: Re: [chensnathan/YOLOF] A problem about the code at train_net.py ? (#23)

@.***

If you have any questions, feel free to drop me an email.

— You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub, or unsubscribe.