ultralytics 8.0.76 minor fixes and improvements (#2004)

Co-authored-by: Seungtaek Kim <seungtaek.kim.94@gmail.com>
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Laughing <61612323+Laughing-q@users.noreply.github.com>
Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: Ercalvez <45692523+Ercalvez@users.noreply.github.com>
Co-authored-by: Erwan CALVEZ <ecalvez@enib.fr>
This commit is contained in:
Glenn Jocher
2023-04-13 20:19:42 +02:00
committed by GitHub
parent 48c4483795
commit 4916014af2
15 changed files with 63 additions and 26 deletions

View File

@ -139,7 +139,7 @@ predicts the classes and locations of objects in the input images or videos.
results = model.predict(source=0, stream=True)
for result in results:
# detection
# Detection
result.boxes.xyxy # box with xyxy format, (N, 4)
result.boxes.xywh # box with xywh format, (N, 4)
result.boxes.xyxyn # box with xyxy format but normalized, (N, 4)
@ -147,12 +147,12 @@ predicts the classes and locations of objects in the input images or videos.
result.boxes.conf # confidence score, (N, 1)
result.boxes.cls # cls, (N, 1)
# segmentation
result.masks.masks # masks, (N, H, W)
# Segmentation
result.masks.data # masks, (N, H, W)
result.masks.xy # x,y segments (pixels), List[segment] * N
result.masks.xyn # x,y segments (normalized), List[segment] * N
# classification
# Classification
result.probs # cls prob, (num_class, )
# Each result is composed of torch.Tensor by default,