|
|
@@ -186,7 +186,7 @@ class DoorInference(object):
|
|
|
|
|
|
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
|
|
|
|
|
|
"""
|
