Chen Gen
affa284352
Refactor/reduce G/C/D/IoU `if: else` statements ( #6087 )
* Refactor the code to reduece else
* Update metrics.py
* Cleanup
Co-authored-by: Cmos <gen.chen@ubisoft.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 years ago
JieLi
afa5cfb0f8
Reduce G/D/CIoU logic operations ( #6074 )
Consider that the default value is CIOU,adjust the order of judgment could reduce the number of judgments.
And “elif CIoU:” didn't need 'if'.
Co-authored-by: 李杰 <360751194@qq.comqq.com>
2 years ago
Glenn Jocher
36d12a500e
Explicitly compute TP, FP in val.py ( #5727 )
3 years ago
NauchtanRobotics
0be58f16f0
Adjust legend labels for classes without instances ( #5174 )
* legend-labels Adjust legend labels for classes without instances.
* #5158 Re-indexed series names: only classes containing data.
* #5158 Re-indexed series names: only classes containing data.
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 years ago
Glenn Jocher
bb5ebc290e
Close `matplotlib` plots after opening ( #4612 )
* Close plots
* Replace fig.close() for plt.close()
3 years ago
Glenn Jocher
8ac96b7975
Fix for AP calculation limits 0.0 - 1.0 ( #4563 )
This PR brings alignment in AP computation practices with Detectron2 and MMDetection.
Problem first noted by @yusiyoh in https://github.com/ultralytics/yolov5/issues/4546
3 years ago
Glenn Jocher
24bea5e4b7
Standardize headers and docstrings ( #4417 )
* Implement new headers
* Reformat 1
* Reformat 2
* Reformat 3 - math
* Reformat 4 - yaml
3 years ago
Glenn Jocher
c6c88dc601
Copy-Paste augmentation for YOLOv5 ( #3845 )
* Copy-paste augmentation initial commit
* if any segments
* Add obscuration rejection
* Add copy_paste hyperparameter
* Update comments
3 years ago
Glenn Jocher
5ea771d93d
Move IoU functions to metrics.py ( #3820 )
3 years ago
Glenn Jocher
e8c5237403
ConfusionMatrix `normalize=True` fix ( #3587 )
3 years ago
Glenn Jocher
ec2da4a82c
Add ConfusionMatrix `normalize=True` flag ( #3586 )
3 years ago
Glenn Jocher
095197bd4a
Ignore Seaborn plot warnings ( #3576 )
* Ignore Seaborn plot warnings
* Update plots.py
* Update metrics.py
3 years ago
Michael Heilig
d48a34dca7
bug fix: switched rows and cols for correct detections in confusion matrix ( #2883 )
3 years ago
Ryan Avery
2c56ad5436
Confusion matrix background axis swap ( #2114 )
3 years ago
Glenn Jocher
f639e14e4d
Metric-Confidence plots feature addition ( #2057 )
* Metric-Confidence plots feature addition
* cleanup
* Metric-Confidence plots feature addition
* cleanup
* Update run-once lines
* cleanup
* save all 4 curves to wandb
3 years ago
Glenn Jocher
0bb43953eb
Reinstate PR curve sentinel values ( #1645 )
3 years ago
Glenn Jocher
f010147578
Update matplotlib.use('Agg') tight ( #1583 )
* Update matplotlib tight_layout=True
* udpate
* udpate
* update
* png to ps
* update
* update
3 years ago
Glenn Jocher
7aeef2dca5
Prevent PR plotting ( #1489 )
4 years ago
Glenn Jocher
0a3ff71ae0
Confusion matrix ( #1474 )
* initial commit
* add plotting
* matrix to cpu
* bug fix
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* seaborn pandas to requirements.txt
* seaborn pandas to requirements.txt
* update wandb plotting
* remove pandas
* if plots
* if plots
* if plots
* if plots
* if plots
* initial commit
* add plotting
* matrix to cpu
* bug fix
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* update plot
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* cleanup
* seaborn pandas to requirements.txt
* seaborn pandas to requirements.txt
* update wandb plotting
* remove pandas
* if plots
* if plots
* if plots
* if plots
* if plots
* Cat apriori to autolabels
* cleanup
4 years ago
Glenn Jocher
4250f84dfb
Update PR curve ( #1428 )
* Update PR curve
* legend outside
* list(Path().glob())
4 years ago
Glenn Jocher
fe341fa44d
Utils reorganization ( #1392 )
* Utils reorganization
* Add new utils files
* cleanup
* simplify
* reduce datasets.py
* remove evolve.sh
* loadWebcam cleanup
4 years ago