From e7c77ee9fe78d8b82caa5b01574792af429fe923 Mon Sep 17 00:00:00 2001 From: Marcos Luciano Date: Sun, 26 Mar 2023 19:00:25 -0300 Subject: [PATCH] Update PPYOLOE conversion --- utils/gen_wts_ppyoloe.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/utils/gen_wts_ppyoloe.py b/utils/gen_wts_ppyoloe.py index 058e705..1c10008 100644 --- a/utils/gen_wts_ppyoloe.py +++ b/utils/gen_wts_ppyoloe.py @@ -418,7 +418,7 @@ with open(wts_file, 'w') as fw, open(cfg_file, 'w') as fc: layers.Shuffle(reshape=[model.yolo_head.num_classes, 'hw'], output='cls') layers.ESEAttn(model.yolo_head.stem_reg[i], route=-7) layers.Conv2D(model.yolo_head.pred_reg[i]) - layers.Shuffle(reshape=[4, reg_max + 1, 'hw'], transpose2=[1, 0, 2]) + layers.Shuffle(reshape=[4, reg_max, 'hw'], transpose2=[1, 0, 2]) layers.SoftMax(0) layers.Conv2D(model.yolo_head.proj_conv) layers.Shuffle(reshape=['h', 'w'], output='reg')