diff --git a/YOLONAS.md b/YOLONAS.md index 709d571888..26cbd8908f 100644 --- a/YOLONAS.md +++ b/YOLONAS.md @@ -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 ```