修改缩进错误
This commit is contained in:
@@ -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