New YOLOv5 conversion and support (>= v2.0)

This commit is contained in:
Marcos Luciano
2022-07-14 13:58:55 -03:00
parent ea31a54bbb
commit 8983e7e99e
2 changed files with 4 additions and 11 deletions

View File

@@ -210,16 +210,13 @@ class Layers(object):
stride = m.stride
size = m.kernel_size
pad = m.padding
mode = m.ceil_mode
m = 'maxpool_up' if mode else 'maxpool'
p = 'pad=%d\n' % pad if pad > 0 else ''
self.fc.write('\n[%s]\n' % m +
'stride=%d\n' % stride +
'size=%d\n' % size +
p)
'size=%d\n' % size)
def upsample(self, child):
self.blocks[self.current] += 1