Consistent saved_model output format (#7032)
This commit is contained in:
1 parent
7c6a33564a
commit
4effd064b1
2 files changed
+2
-2
No files matched your search
+1
-1
@@ -441,7 +441,7 @@ class DetectMultiBackend(nn.Module):
|
||||
else: # TensorFlow (SavedModel, GraphDef, Lite, Edge TPU)
|
||||
im = im.permute(0, 2, 3, 1).cpu().numpy() # torch BCHW to numpy BHWC shape(1,320,192,3)
|
||||
if self.saved_model: # SavedModel
|
||||
y = (self.model(im, training=False) if self.keras else self.model(im)[0]).numpy()
|
||||
y = (self.model(im, training=False) if self.keras else self.model(im)).numpy()
|
||||
elif self.pb: # GraphDef
|
||||
y = self.frozen_func(x=self.tf.constant(im)).numpy()
|
||||
else: # Lite or Edge TPU
|
||||
|
||||
Reference in new issue
Block a user