search

lyuwenyu / RT-DETR

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥
Apache License 2.0
2.57k stars 298 forks source link

将epoch的值引入RT-DETR的head中 #426

Open sdreamforchen opened 2 months ago

sdreamforchen commented 2 months ago

您好，想请教一下，如何将训练的epoch/iteration配置到RT-DETR的head中呢？ 类似output=αA+（1-α）B，α=epoch/total_epoch；因为现有的architectures架构是将backbone、neck和head一起传到train里面的，不是很懂如何单独对head进行操作，且能传train中的epoch/iteration参数；能否给一个示例，谢谢！

lyuwenyu commented 2 months ago

把head的forward改造成可以接收epoch的参数；同理 改造模型forward；然后在训练过程中 传入就行吧