Compare commits
1 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
fbc5cf49d8 |
@@ -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())
|
||||
|
||||
|
||||
Reference in New Issue
Block a user