MobileViTv2Attention 是否可以整合到 yolov8 中?
作者: dahaigui创建于 2024年10月16日更新于 2024年10月16日
我想把 https://GitHub.com/xmu-xiaoma666/External-Attention-pytorch/blob/master/model/attention/MobileViTv2Attention.py 整合到 yolov8 中,将 MobileViTv2Attention 当作一个模块使用,但配置了很长时间,发现无法调整输入和输出,有没有大佬可以帮忙操作一下?
内容来源: xmu-xiaoma666/External-Attention-pytorch