Bobo-y / flexible-yolov5

More readable and flexible yolov5 with more backbone(gcn, resnet, shufflenet, moblienet, efficientnet, hrnet, swin-transformer, etc) and (cbam,dcn and so on), and tensorrt

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Can't I use another pre-trained model after changing the backbone?

yuhao-q opened this issue · comments

commented

of course, such as hwen you use resnet-bakcbone, you can't use yolov5 pre-trained model

Thank you for your answer. What I want to ask is, if I use swin Tiny as the backbone, can I use swin_tiny_patch4_window7_224.pth as the pretrained model, I tried it just now and found it failed

commented

Thank you for your answer. What I want to ask is, if I use swin Tiny as the backbone, can I use swin_tiny_patch4_window7_224.pth as the pretrained model, I tried it just now and found it failed

yes

commented

i will check it