如何在yolov5中添加stn（空间转换网络）模块呀，像添加注意力机制那样？

博主你好，在yolov5中添加stn（空间转换网络）模块呀，像添加注意力机制那样