Skip to content

Commit

Permalink
Fix invalid syntax at YOLONAS.md (#1441)
Browse files Browse the repository at this point in the history
  • Loading branch information
Pslydhh committed Sep 1, 2023
1 parent 8f5b0a8 commit 492aef7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion YOLONAS.md
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ prediction = model_predictions[0].prediction # One prediction per image -

bboxes = prediction.bboxes_xyxy # [[Xmin,Ymin,Xmax,Ymax],..] list of all annotation(s) for detected object(s)
bboxes = prediction.bboxes_xyxy # [[Xmin,Ymin,Xmax,Ymax],..] list of all annotation(s) for detected object(s)
class_names = prediction.class_names. # ['Class1', 'Class2', ...] List of the class names
class_names = prediction.class_names # ['Class1', 'Class2', ...] List of the class names
class_name_indexes = prediction.labels.astype(int) # [2, 3, 1, 1, 2, ....] Index of each detected object in class_names(corresponding to each bounding box)
confidences = prediction.confidence.astype(float) # [0.3, 0.1, 0.9, ...] Confidence value(s) in float for each bounding boxes
```
Expand Down

0 comments on commit 492aef7

Please sign in to comment.