yolov8 snpe Converter does not support ‘expand‘ op type

最新推荐文章于 2024-04-12 17:46:05 发布

AI算法网奇

最新推荐文章于 2024-04-12 17:46:05 发布

阅读量297

点赞数 1

分类专栏： yolo pytorch知识宝典文章标签： YOLO pytorch 深度学习

本文链接：https://blog.csdn.net/jacke121/article/details/133581567

版权

pytorch知识宝典同时被 2 个专栏收录

499 篇文章 226 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

yolo

68 篇文章 27 订阅

订阅专栏

在将YOLOv8模型从PyTorch转换时遇到错误，Converter不支持'expand'操作类型。解决方法是由于模型分辨率固定，可以预先生成anchors并在后续步骤中重复使用。

摘要由CSDN通过智能技术生成

Converter does not support 'expand' op type

后来发现分辨率一旦固定，anchors也就固定了，提前生成anchors，后面调用就行了

Converter does not support 'expand' op type

原来报错的代码：

    anchor_points, stride_tensor = [], []
    h_s=[[48,80],[24,40],[12,20]]
    dtype, device = feats[0].dtype, feats[0].device
    for i, stride in enumerate(strides):
        h, w = h_s[i]
        sx = torch.arange(end=w, device=device, dtype=dtype) + grid_cell_offset  # shift x
        sy = torch.arange(end=h, device=device, dtype=dtype) + grid_cell_offset  # shift y
        sy, sx = torch.meshgrid(sy, sx, indexing='ij') if TORCH_1_10 else torch.meshgrid(sy, sx)
        anchor_points.append(torc