Skip to content

Commit 77977e0

Browse files
authored
CoreML inference fix list() -> sorted() (#6496)
1 parent 5e4ff19 commit 77977e0

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

models/common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -427,7 +427,7 @@ def forward(self, im, augment=False, visualize=False, val=False):
427427
conf, cls = y['confidence'].max(1), y['confidence'].argmax(1).astype(np.float)
428428
y = np.concatenate((box, conf.reshape(-1, 1), cls.reshape(-1, 1)), 1)
429429
else:
430-
y = y[list(y)[-1]] # last output
430+
y = y[sorted(y)[-1]] # last output
431431
else: # TensorFlow (SavedModel, GraphDef, Lite, Edge TPU)
432432
im = im.permute(0, 2, 3, 1).cpu().numpy() # torch BCHW to numpy BHWC shape(1,320,192,3)
433433
if self.saved_model: # SavedModel

0 commit comments

Comments
 (0)