有关head的问题
cds221 opened this issue · comments
请问yolo head能否换成基于自注意力机制的检测头呢
可以
请问大佬您知道有没有适合yolov5的自注意力机制的head吗？
没太了解呢
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