1 Commits

Author SHA1 Message Date
Kevin Wong
fbc5cf49d8 修改缩进错误 2026-01-06 10:47:41 +08:00

View File

@@ -271,6 +271,7 @@ class IndoorNavigator:
masks = r0.masks.data.cpu().numpy()
boxes = r0.boxes
for i, (mask, cls_id, conf) in enumerate(zip(masks, boxes.cls, boxes.conf)):
cls_id = int(cls_id.item())
conf_val = float(conf.item())