使用tensorrtx编写的NAM注意力模块
Primary LanguageC++Apache License 2.0Apache-2.0
common.hpp文件使用的是文件地址里面的yolov5目录的。
common.hpp