HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Add YOLOv5 dataset yamls (#207)

Browse source 
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
Glenn Jocher 2023-01-10 18:33:44 +01:00 • committed by GitHub
parent e371e81aa0
commit c7629e93bd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
14 changed files with 2018 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ultralytics/yolo/v8/classify/predict.py
View file

@ -38,7 +38,7 @@ class ClassificationPredictor(BasePredictor):
log_string += '%gx%g ' % im.shape[2:] # print string
self.annotator = self.get_annotator(im0)
prob = preds[idx]
prob = preds[idx].softmax(0)
self.all_outputs.append(prob)
# Print results
top5i = prob.argsort(0, descending=True)[:5].tolist() # top 5 indices