Philip Gutjahr
529fbc1814
Use PIL to eliminate chroma subsampling in crops ( #7008 )
* use pillow to save higher-quality jpg (w/o color subsampling)
* Cleanup and doc issue
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 år sedan
Max Strobel
7c6a33564a
fix: add default PIL font as fallback ( #7010 )
* fix: add default font as fallback
Add default font as fallback if the downloading of the Arial.ttf font
fails for some reason, e.g. no access to public internet.
* Update plots.py
Co-authored-by: Maximilian Strobel <Maximilian.Strobel@infineon.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 år sedan
Glenn Jocher
c21da596f3
Fix `hyp_evolve.yaml` indexing bug ( #6604 )
* Fix `hyp_evolve.yaml` indexing bug
Bug caused hyp_evolve.yaml to display latest generation result rather than best generation result.
* Update plots.py
* Update general.py
* Update general.py
* Update general.py
2 år sedan
Glenn Jocher
f627bc53f8
Fix `plot_labels()` colored histogram bug ( #6574 )
* Fix `plot_labels()` colored histogram bug
* Cleanup
2 år sedan
Jirka Borovec
cba4303d32
Fix 6 Flake8 issues ( #6541 )
* F541
* F821
* F841
* E741
* E302
* E722
* Apply suggestions from code review
* Update general.py
* Update datasets.py
* Update export.py
* Update plots.py
* Update plots.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2 år sedan
Glenn Jocher
c3e599cfda
Update val_batch*.jpg for Chinese fonts ( #6526 )
* Update plots for Chinese fonts
* make is_chinese() non-str safe
* Add global FONT
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update general.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2 år sedan
Glenn Jocher
d5966c93f1
Namespace `VERBOSE` env variable to `YOLOv5_VERBOSE` ( #6428 )
* Verbose updates
* Verbose updates
2 år sedan
Glenn Jocher
a3d5f1d3e3
Revert "Update `plot_lr_scheduler()` ( #5864 )" ( #5920 )
This reverts commit 360eec6910
.
2 år sedan
Can
360eec6910
Update `plot_lr_scheduler()` ( #5864 )
shallow copy modify originals
2 år sedan
Zengyf-CVer
f17c86b7f0
Save *.npy features on detect.py `--visualize` ( #5701 )
* Add feature map to save npy files
Add feature map to save npy files,export npy files with 32 feature maps per layer.
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update plots.py
* Update plots.py
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update plots.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 år sedan
Glenn Jocher
80cfaf40ef
AutoAnchor and AutoBatch `LOGGER` ( #5635 )
* AutoBatch, AutoAnchor `LOGGER`
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update autoanchor.py
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
3 år sedan
Glenn Jocher
def7a0fd19
Improve plots.py robustness ( #5616 )
* Improve plots.py robustness
Addresses issues #5374, #5395, #5611
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
3 år sedan
Glenn Jocher
3f64ad1760
Fix `save_one_box()` ( #5545 )
* Fix `save_one_box()`
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
3 år sedan
Jirka Borovec
0155548384
precommit: isort ( #5493 )
* precommit: isort
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update isort config
* [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
* Update name
Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 år sedan
Glenn Jocher
5866646cc8
Fix float zeros format ( #5491 )
* Fix float zeros format
* 255 to integer
3 år sedan
Jirka Borovec
ed887b5976
Add pre-commit CI actions ( #4982 )
* define pre-commit
* add CI code
* configure
* apply pre-commit
* fstring
* apply MD
* pre-commit
* Update torch_utils.py
* Update print strings
* notes
* Cleanup code-format.yml
* Update setup.cfg
* Update .pre-commit-config.yaml
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 år sedan
Glenn Jocher
4cf7d487a6
Update plots.py ( #5127 )
3 år sedan
Diego Montes
c1bed601e9
Fix `isascii()` method calls for python 3.6 ( #4958 )
* fix isascii for python3.6
* update comment with python 3.7 note
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 år sedan
Glenn Jocher
a820b43aca
Automatic Chinese fonts plotting ( #4951 )
* Automatic Chinese fonts plotting
* Default PIL=False
3 år sedan
Jiacong Fang
ce7fa81d4e
Avoid out-of-image class labels ( #4902 )
* Avoid out-of-image class labels
* Update plots.py
* Cleanup
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 år sedan
Glenn Jocher
8ad9e4ed5b
Annotator `check_font()` RANK -1 remove progress ( #4864 )
* Annotator `check_font()` RANK -1 remove progress
* Cleanup
3 år sedan
Glenn Jocher
3a822a22ce
`PIL.ImageDraw.text(anchor=...)` removal, reduce to `>=7.1.2` ( #4842 )
* Unpin Pillow
* Update requirements.txt
* Update plots.py
3 år sedan
Glenn Jocher
27a4736e96
Fix val.py study plot ( #4831 )
* Fix val.py study plot
* call plot_val_study
* Rename plot_study_txt to plot_val_study
3 år sedan
Glenn Jocher
cd810c8286
Centralize `user_config_dir()` decision making ( #4755 )
3 år sedan
JD Costa
deb434aefa
Make CONFIG_DIR configurable per environment variable ( #4727 )
3 år sedan
Glenn Jocher
8e94bf62d9
Add `user_config_dir('Ultralytics')` ( #4715 )
* Add `user_config_dir`
* Linux to .config
3 år sedan
Glenn Jocher
234e8ae6fa
Fix for `plot_evolve()` string argument ( #4639 )
3 år sedan
Glenn Jocher
de534e9221
Fix 3 for Arial.ttf redownloads with hub inference ( #4629 )
Fix 3 for Arial.ttf redownloads with hub inference, follow-on to #4628 .
3 år sedan
Glenn Jocher
a4e8f78c5e
Fix 2 for Arial.ttf redownloads with hub inference ( #4628 )
3 år sedan
Glenn Jocher
bb5ebc290e
Close `matplotlib` plots after opening ( #4612 )
* Close plots
* Replace fig.close() for plt.close()
3 år sedan
Glenn Jocher
e7d1842a98
Auto-download Arial.ttf on init ( #4606 )
* Auto-download Arial.ttf on init
* Fix ROOT
3 år sedan
Glenn Jocher
b894e69dfc
Update mosaic plots font size ( #4596 )
3 år sedan
Glenn Jocher
dbbc6b5c48
Re-order `plots.py` to class-first ( #4595 )
3 år sedan
Glenn Jocher
de44376d1b
Create `Annotator()` class ( #4591 )
* Add Annotator() class
* Download Arial
* 2x for loop
* Cleanup
* tuple 2 list
* max_size=1920
* bold logging results to
* tolist()
* im = annotator.im
* PIL save in detect.py
* Smart asarray in detect.py
* revert to cv2.imwrite
* Cleanup
* Return result asarray
* Add `Profile()` profiler
* CamelCase Timeout
* Resize after mosaic
* pillow>=8.0.0
* daemon imwrite
* Add cv2 support
* Remove plot_wh_methods and plot_one_box
* pil=False for hubconf.py annotations
* im.shape bug fix
* colorstr common.py
* join daemons
* Update t.daemon
* Removed daemon saving
3 år sedan
Takumi Karasawa
7b35971ba5
Fix bug in `plot_one_box` when label is `None` ( #4588 )
3 år sedan
Glenn Jocher
24bea5e4b7
Standardize headers and docstrings ( #4417 )
* Implement new headers
* Reformat 1
* Reformat 2
* Reformat 3 - math
* Reformat 4 - yaml
3 år sedan
Glenn Jocher
2da4e7acf7
Merge PIL and OpenCV in `plot_one_box(use_pil=False)` ( #4416 )
* Merge PIL and OpenCV box plotting functions
* Add ASCII check to plot_one_box
* Cleanup
* Cleanup2
3 år sedan
Glenn Jocher
e086347377
Update plots.py ( #4407 )
3 år sedan
Glenn Jocher
e78aeac973
Evolve in CSV format ( #4307 )
* Update evolution to CSV format
* Update
* Update
* Update
* Update
* Update
* reset args
* reset args
* reset args
* plot_results() fix
* Cleanup
* Cleanup2
3 år sedan
Kalen Michael
b74929c910
Add `train.py` and `val.py` callbacks ( #4220 )
* added callbacks
* Update callbacks.py
* Update train.py
* Update val.py
* Fix CamlCase add staticmethod
* Refactor logger into callbacks
* Cleanup
* New callback on_val_image_end()
* Add curves and results images to TensorBoard
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 år sedan
Glenn Jocher
96e36a7c91
New CSV Logger ( #4148 )
* New CSV Logger
* cleanup
* move batch plots into Logger
* rename comment
* Remove total loss from progress bar
* mloss :-1 bug fix
* Update plot_results()
* Update plot_results()
* plot_results bug fix
3 år sedan
Glenn Jocher
efe60b5681
Refactor train.py and val.py `loggers` ( #4137 )
* Update loggers
* Config
* Update val.py
* cleanup
* fix1
* fix2
* fix3 and reformat
* format sweep.py
* Logger() class
* cleanup
* cleanup2
* wandb package import fix
* wandb package import fix2
* txt fix
* fix4
* fix5
* fix6
* drop wandb into utils/loggers
* fix 7
* rename loggers/wandb_logging to loggers/wandb
* Update message
* Update message
* Update message
* cleanup
* Fix x axis bug
* fix rank 0 issue
* cleanup
3 år sedan
Glenn Jocher
720aaa65c8
Rename `test.py` to `val.py` ( #4000 )
3 år sedan
Glenn Jocher
248504cf13
Feature visualization improvements 32 ( #3947 )
3 år sedan
Glenn Jocher
850970e081
`feature_visualization()` CUDA fix ( #3925 )
3 år sedan
Glenn Jocher
87b094bcbc
Feature visualization update ( #3920 )
* Feature visualization update
* Save to jpg (faster)
* Save to png
3 år sedan
Glenn Jocher
02719dde52
Update `feature_visualization()` ( #3807 )
* Update `feature_visualization()`
Only plot for data with height, width > 1
* cleanup
* Cleanup
3 år sedan
Zigarss
20d45aa4f1
Add feature map visualization ( #3804 )
* Add feature map visualization
Add a feature_visualization function to visualize the mid feature map of the model.
* Update yolo.py
* remove boolean from forward and reorder if statement
* remove print from forward
* General cleanup
* Indent
* Update plots.py
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
3 år sedan
Glenn Jocher
095197bd4a
Ignore Seaborn plot warnings ( #3576 )
* Ignore Seaborn plot warnings
* Update plots.py
* Update metrics.py
3 år sedan
Glenn Jocher
5948f20a3d
Update test.py profiling ( #3555 )
* Update test.py profiling
* half_precision to half
* inplace
3 år sedan