如何在yolov8中使用ATSS策略
在yolov8中使用的标签匹配策略是TAL,本篇文章解析一下ATSS代码相关实现以及如何把ATSS放到yolov8中使用
看过本专栏中的另外两篇文章的同学应该对v8解析box那一套很熟悉了，ATSS的第一步就是去得到一系列的anchor-box(如果是anchor-based检测方法)或者anchor-point(基于anchor-free的方法)
这里首先在v8代码目录ultralytics/utils/下增加ATSS的实现,然后增加生成锚点的方法
def generate_anchors_free(feats, fpn_strides, grid_cell_size=5.0, grid_cell_offset=0.5, device='cpu', is_eval=False, mode='af'):
'''Generate anchors from features.'''
anchors = []
anchor_points = []
stride_tensor = []
num_anchors_list = []
assert feats is not None
if is_eval:
for i, stride in enumerate(fpn_strides):
_, _, h, w 原文地址：https://blog.csdn.net/luoshiyong123/article/details/144776370
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！