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.31k stars 259 forks source link

How do I set up not to use pre-trained models in the backbone network #246

Open wsy-yjys opened 6 months ago

wsy-yjys commented 6 months ago

How do I set up not to use pre-trained models in the backbone network?

lyuwenyu commented 6 months ago

没太理解具体什么需求


或许这些地方是你需要的

  1. https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetr_pytorch/src/solver/solver.py#L96
  2. https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetr_pytorch/src/nn/backbone/presnet.py#L142
  3. https://github.com/lyuwenyu/RT-DETR/blob/main/rtdetr_pytorch/configs/rtdetr/include/optimizer.yml#L18
wsy-yjys commented 6 months ago

好的,谢谢

KoniHD commented 2 months ago

Sorry for not speaking Chinese and being new to ML.

Just simply curious. Does the model automatically use pre-trained weights if not specified otherwise or does it start from scratch with randomly initialized weights?

Help would be much appreciated.