thebears/deepstream_yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update PPYOLOE conversion

Browse Source
This commit is contained in:
Marcos Luciano
2023-03-26 19:00:25 -03:00
parent 1a7dce1e47
commit e7c77ee9fe
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
utils/gen_wts_ppyoloe.py
View File

@@ -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.Shuffle(reshape=[model.yolo_head.num_classes, 'hw'], output='cls')
layers.ESEAttn(model.yolo_head.stem_reg[i], route=-7) layers.ESEAttn(model.yolo_head.stem_reg[i], route=-7)
layers.Conv2D(model.yolo_head.pred_reg[i]) 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.SoftMax(0)
layers.Conv2D(model.yolo_head.proj_conv) layers.Conv2D(model.yolo_head.proj_conv)
layers.Shuffle(reshape=['h', 'w'], output='reg') layers.Shuffle(reshape=['h', 'w'], output='reg')