xclele/yolov10
1
0
Fork 0
mirror of https://github.com/THU-MIG/yolov10.git synced 2025-05-23 13:34:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
1,096 Commits 1 Branch 2 Tags
Commit Graph

1096 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
Unai Gurbindo
af057b0c4e
Merge a974079bd5b867303b7c77442be14c11ca312482 into 453c6e38a51e9d1d5a2aa5fb7f1014a711913397 2025-03-15 18:57:04 +08:00
Wang Ao
453c6e38a5
Update README.md 2025-03-14 10:53:57 +08:00
Wang Ao
475f7c7365
Update README.md 2025-03-13 20:08:35 +08:00
Unai Gurbindo
a974079bd5
Update metrics.py 
## Possible error in the `ap_per_class` function when calculating the precision-recall curve

Hello,

I have identified a possible error in the `ap_per_class` function when calculating the metrics for the precision-recall curve. Below, I provide an example to explain it more clearly:

Let’s assume the `unique_classes` list is `[0, 1, 2]`, and the model never predicted class 1. In this case, on line 583 of the `utils/metrics.py` file, the loop will skip to the next class (class 2) because `n_p == 0` for class 1. 

However, before moving on to the next class, a zero-filled array corresponding to class 1 should be added to the `prec_values` list. Otherwise, the precision-recall curve for class 2 will incorrectly appear as if it belongs to class 1, and the legend will also fail to show the correct label for class 2.

To fix this behavior, before the `continue` statement, the code should be modified as follows:

```python
if n_p == 0 or n_l == 0:
    prec_values.append(np.zeros(1000))
    continue
```

I hope my explanation is clear, and I have correctly identified the issue. While this is a rare case where the model never predicts one of the classes in the dataset, it is still something to consider to avoid errors in the precision-recall curve visualizations.

Best regards.
 2024-10-08 12:32:30 +02:00
Wang Ao
6fbaf42b23
Update README.md 2024-09-26 19:12:57 +08:00
inisis
cd2f79c702
fix onnxslim version (#380) 2024-07-30 15:16:55 +08:00
leonnil
0e51495ce6 replace onnxsim with onnxslim 2024-07-30 04:34:28 +00:00
leonnil
aad320dd80 prediction augment (TTA) fix (#288) 2024-06-23 08:21:40 +00:00
leonnil
2c36ab0f10 ddp resume checkpoint fix (#184) 2024-06-15 16:22:32 +00:00
Wang Ao
36efe34fe1
update README 2024-06-12 13:28:44 +08:00
shiraha721
3a53d91ec8
fix cv2.error when use PIL.Image.Image (#241) 2024-06-11 21:55:42 +08:00
wa22
ea93d4f379 update huggingface hub card template 2024-06-07 16:22:42 +08:00
wa22
e161d70ff7 update huggingface hub card template 2024-06-07 16:19:04 +08:00
Pbatch
d8777c1449
Make postprocessing compatible with nobuco (#204) 2024-06-06 07:57:59 +08:00
Ankan Ghosh
13f6ab770d
Update README.md (#198) 2024-06-05 21:33:23 +08:00
kangsanha
e6d80f3fa6
Update __init__.py (#191) 
/Desktop/ultralytics/runs/detect/ -> /Desktop/yolov10/runs/detect/
 2024-06-04 19:19:53 +08:00
kangsanha
15bdea757a
fix onnx model val.py AssertionError (#190) 2024-06-04 19:18:04 +08:00
leonnil
ba0e6cbc1a fix onnx validation (#179) 2024-06-04 10:29:21 +00:00
wa22
e5f2f03c55 use yolov10 by default in cli 2024-06-04 11:06:48 +08:00
wa22
dfed201384 add support for HuggingFace Hub for cli 2024-06-03 23:04:27 +08:00
Ekaterina Aidova
060af8b8c1
Small fix in detection head for enabling ov export from scratch (#183) 2024-06-03 17:59:44 +08:00
Sencer Yücel
1cfe7a4e13
Added video output support for gradio (#113) 
added video output support for gradio

---------

Co-authored-by: Sencer Yücel <sencer.yucel@turkai.com>
Co-authored-by: wa22 <wa22@mails.tsinghua.edu.cn>
 2024-06-03 17:15:31 +08:00
wa22
1539b5a678 add calculation for flops 2024-06-03 11:21:49 +08:00
wa22
587981fc27 update app.py 2024-06-03 10:50:08 +08:00
NielsRogge
a6e064d331
Track downloads (#172) 
Track downloads from the HuggingFace hub
 2024-06-03 10:23:37 +08:00
wa22
7a0cce17b6 add names when pushing to huggingface hub 2024-06-03 10:21:43 +08:00
Erlangga Yudi Pradana
0efdd2f0c4
docs: update README.md (#174) 2024-06-01 22:48:36 +08:00
wa22
22f49e1249 update requirements.txt 2024-06-01 21:32:44 +08:00
wa22
16676546f1 update README 2024-06-01 21:32:27 +08:00
wa22
8d0e567e58 update README 2024-06-01 20:18:30 +08:00
wa22
0f87aa3173 Merge branch 'main' of github.com:jameslahm/yolov10 into main 2024-06-01 19:12:41 +08:00
wa22
43c4a76043 update requirements.txt 2024-06-01 19:11:00 +08:00
NielsRogge
989cdb812d
Add mixin (#168) 2024-06-01 19:05:46 +08:00
wa22
b70277e940 update README 2024-06-01 18:57:08 +08:00
wa22
22aa8412a7 update README 2024-06-01 10:53:02 +08:00
wa22
3594353249 update README 2024-05-31 18:33:16 +08:00
yuyoujiang
393091b63e
update README (#157) 2024-05-31 17:02:14 +08:00
mohamedsamirx
c829350969
Update README.md (#152) 2024-05-31 12:57:49 +08:00
wa22
6d402a68e1 Merge branch 'main' of github.com:jameslahm/yolov10 into main 2024-05-31 10:55:17 +08:00
wa22
1788e9d4e8 update README 2024-05-31 10:55:07 +08:00
Veeja Liu
634ceee537
Change the format of README.md (#137) 
Co-authored-by: weijialiu <weijialiu@hireez.com>
 2024-05-31 10:27:26 +08:00
wa22
27a889edbd update README 2024-05-31 09:26:52 +08:00
wa22
bbcad68d77 update README 2024-05-31 09:25:17 +08:00
wa22
dadcda9c70 update README 2024-05-30 09:50:02 +00:00
wa22
2b24dab60f update README 2024-05-30 09:49:11 +00:00
wa22
842e8ababe update README 2024-05-30 09:43:13 +00:00
wa22
8f0984817d fix val with half 2024-05-30 15:46:16 +08:00
wa22
cf32e2f7f0 update README 2024-05-29 14:22:38 +08:00
wa22
718c899e7e update README 2024-05-29 13:53:09 +08:00
wa22
48af6f097d update README 2024-05-29 13:08:55 +08:00