Merge pull request #416 from roboflow/notebook-fixes

Supervision demo notebook fix and yolo-nas fix
This commit is contained in:
Piotr Skalski authored and GitHub committed 2023-10-05 15:18:13 +02:00
commit 77c126e421
3 files changed
+530 -529

No files matched your search

+1
View File
@@ -22,6 +22,7 @@ repos:
- id: detect-private-key
- id: forbid-new-submodules
- id: pretty-format-json
exclude: demo.ipynb
args: ['--autofix', '--no-sort-keys', '--indent=4']
- id: end-of-file-fixer
- id: mixed-line-ending
Vendored
+528 -528
View File
File diff suppressed because it is too large. Load diff
+1 -1
View File
@@ -296,7 +296,7 @@ class Detections:
>>> detections = sv.Detections.from_yolo_nas(result)
```
"""
if np.asarray(yolo_nas_results.bboxes_xyxy).shape[0] == 0:
if np.asarray(yolo_nas_results.prediction.bboxes_xyxy).shape[0] == 0:
return cls.empty()
return cls(