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

Add PP-YOLOE+ support

Browse Source
This commit is contained in:
Marcos Luciano
2023-01-31 02:59:56 -03:00
parent 825d6bfda8
commit 69f29f8934
4 changed files with 61 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
utils/gen_wts_ppyoloe.py
View File

@@ -3,7 +3,7 @@ import struct
import paddle
import numpy as np
from ppdet.core.workspace import load_config, merge_config
from ppdet.utils.check import check_gpu, check_version, check_config
from ppdet.utils.check import check_version, check_config
from ppdet.utils.cli import ArgsParser
from ppdet.engine import Trainer
from ppdet.slim import build_slim_model
@@ -273,13 +273,14 @@ class Layers(object):
def get_state_dict(self, state_dict):
for k, v in state_dict.items():
vr = v.reshape([-1]).numpy()
self.fw.write('{} {} '.format(k, len(vr)))
for vv in vr:
self.fw.write(' ')
self.fw.write(struct.pack('>f', float(vv)).hex())
self.fw.write('\n')
self.wc += 1
if 'alpha' not in k:
vr = v.reshape([-1]).numpy()
self.fw.write('{} {} '.format(k, len(vr)))
for vv in vr:
self.fw.write(' ')
self.fw.write(struct.pack('>f', float(vv)).hex())
self.fw.write('\n')
self.wc += 1
def get_anchors(self, anchor_points, stride_tensor):
vr = anchor_points.numpy()