Update end2end.py

furyhawk · Sep 14, 2022 · 7420f55 · 7420f55
1 parent 16301c0
commit 7420f55
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/yolov6/models/end2end.py b/yolov6/models/end2end.py
@@ -255,6 +255,7 @@ def forward(self, x):
         if self.with_preprocess:
             x = x[:,[2,1,0],...]
             x = x * (1/255)
+        x = self.model(x)
         if isinstance(x, list):
             x = x[0]
         else: