Fix route layer

2022-08-16 14:36:52 -03:00
parent ab082fc292
commit dc023b308e
6 changed files with 33 additions and 21 deletions
--- a/utils/gen_wts_yoloV7.py
+++ b/utils/gen_wts_yoloV7.py
@@ -135,7 +135,8 @@ class Layers(object):
        self.fc.write('[net]\n' +
                      'width=%d\n' % self.width +
                      'height=%d\n' % self.height +
-                      'channels=3\n')
+                      'channels=3\n' +
+                      'letter_box=1\n')

    def reorg(self):
        self.blocks[self.current] += 1