@@ -186,7 +186,7 @@ class DoorInference(object):
# cv2.waitKey(30000)
res = self.door_model(door_image)
class_idx = res[0].probs.top1
- class_conf = res[0].probs.top1conf.cpu().numpy()
+ class_conf = res[0].probs.top1conf.cpu().numpy().item()
return class_idx == 1, class_conf
"""
