ncnn yolov5Focus 算子替换

7z5jn7bk  于 4个月前  发布在  其他
关注(0)|答案(4)|浏览(75)

expectation | 诉求 | 期待する

  1. speed
  2. precision

model | 模型 | モデル

  1. model.param and model.bin

detail | 详细描述 | 詳細な説明

hi nihui:
我看了ncnn reorg 算子的实现,请问 mode 不为 0 时,这种重排列 和 yolov5Focus 里面的像素重排列是一致的吧。yolox-nano 模型, int8 量化的时候,除了修改并加入自定义层 Yolov5Focus方法,是不是也可以把等效它换成 reorg算子.

3phpmpom

3phpmpom1#

Reorg 的输出 channel 顺序和 yolov5focus 的顺序不一样

量化时把这个改为 Reorg 可能对结果影响不大,具体得实践下..

gmol1639

gmol16392#

不过也可以给 Reorg 加一个 yolov5 mode,这样就不用外面注册了。。。

vyu0f0g1

vyu0f0g13#

不过也可以给 Reorg 加一个 yolov5 mode,这样就不用外面注册了。。。

嗯嗯 对的,我自己在这边算的,reorg mode != 0 时,通道排列方式和 YoloV5Focus 是一致的,我试下吧,谢谢nihui哈

zujrkrfu

zujrkrfu4#

不过也可以给 Reorg 加一个 yolov5 mode,这样就不用外面注册了。。。

嗯嗯 对的,我自己在这边算的,reorg mode != 0 时,通道排列方式和 YoloV5Focus 是一致的,我试下吧,谢谢nihui哈

单纯的把 Yolov5Focus 改成 reorg 算子 stride=2, model=1,检测框的置信度并不一样.

相关问题